| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| H A D | reshape.py | 15 """Reshape bijectors."""
35 "Reshape",
47 class Reshape(bijector_lib.Bijector): class in inherits:bijector_lib.Bijector
57 * The `Reshape` bijector automatically broadcasts over the leftmost
68 r = tfd.bijectors.Reshape(event_shape_out=[1, -1])
91 """Creates a `Reshape` bijector.
131 super(Reshape, self).__init__(is_constant_jacobian=True,
180 """Reshape only the event_shape of an input `Tensor`."""
286 # In this particular case of the Reshape bijector, the
|
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | reshape_util.cc | 34 void Reshape(OpKernelContext *context, const Tensor &input_indices_in, function in namespace:tensorflow
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/bijectors/ |
| H A D | reshape_test.py | 15 """Tests for Reshape Bijector."""
23 from tensorflow.contrib.distributions.python.ops.bijectors.reshape import Reshape namespace
58 bijector = Reshape(
84 bijector = Reshape(
110 bijector = Reshape(
132 bijector = Reshape(
148 bijector = Reshape(
163 bijector = Reshape(
181 bijector = Reshape(
199 bijector = Reshape(
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/testing/ |
| H A D | parse_testdata.cc | 298 class Reshape : public Message { class in namespace:tflite::testing
300 explicit Reshape(TestRunner* test_runner) : test_runner_(test_runner) { function in class:tflite::testing::Reshape
339 return Store(new Reshape(test_runner_));
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/applications/ |
| H A D | mobilenet.py | 91 from tensorflow.python.keras._impl.keras.layers import Reshape namespace
501 x = Reshape(shape, name='reshape_1')(x)
505 x = Reshape((classes,), name='reshape_2')(x)
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| H A D | operator.cc | 456 class Reshape class in namespace:toco::tflite
806 ops.emplace_back(new Reshape(::tflite::BuiltinOperator_RESHAPE,
|
| /external/tensorflow/tensorflow/contrib/specs/python/ |
| H A D | specs_ops.py | 111 Reshape = Fun(array_ops.reshape) variable
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/layers/ |
| H A D | core.py | 319 @tf_export('keras.layers.Reshape')
320 class Reshape(Layer): class in inherits:Layer
341 model.add(Reshape((3, 4), input_shape=(12,)))
346 model.add(Reshape((6, 2)))
350 model.add(Reshape((-1, 2, 2)))
356 super(Reshape, self).__init__(**kwargs)
421 base_config = super(Reshape, self).get_config()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | literal_test_util.cc | 756 /* static */ std::unique_ptr<Literal> LiteralTestUtil::Reshape( function in class:xla::LiteralTestUtil
|
| /external/tensorflow/tensorflow/contrib/keras/api/keras/layers/ |
| H A D | __init__.py | 71 from tensorflow.python.keras._impl.keras.layers.core import Reshape namespace
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| H A D | __init__.py | 74 from tensorflow.python.keras._impl.keras.layers.core import Reshape namespace
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| H A D | literal_util.cc | 736 StatusOr<std::unique_ptr<Literal>> Literal::Reshape( function in class:xla::Literal
739 return InvalidArgument("Reshape does not support tuples.");
756 "Shapes before and after Literal::Reshape have different numbers "
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | local_computation_builder.cc | 345 ComputationDataHandle LocalComputationBuilder::Reshape( function in class:xla::swig::LocalComputationBuilder
349 return builder_.Reshape(operand, dimensions, new_sizes);
|
| H A D | xla_client.py | 668 def Reshape(self, operand, dimensions, new_sizes): member in class:ComputationBuilder
669 """Reshape op."""
671 self._client.Reshape(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | computation_builder.cc | 377 ComputationDataHandle ComputationBuilder::Reshape( function in class:xla::ComputationBuilder
393 ComputationDataHandle ComputationBuilder::Reshape( function in class:xla::ComputationBuilder
406 return Reshape(operand, dimensions, new_sizes);
457 return Reshape(operand, new_sizes);
|