| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| reshape.py | 15 """Reshape bijectors."""
35 "Reshape",
47 class Reshape(bijector_lib.Bijector):
50 The semantics generally follow that of `tf.reshape()`, with
57 * The `Reshape` bijector automatically broadcasts over the leftmost
60 number of dimensions to reshape is inferred from the provided
68 r = tfd.bijectors.Reshape(event_shape_out=[1, -1])
91 """Creates a `Reshape` bijector.
111 with ops.name_scope(name, "reshape",
131 super(Reshape, self).__init__(is_constant_jacobian=True
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| reshape_util.cc | 34 void Reshape(OpKernelContext *context, const Tensor &input_indices_in,
83 errors::InvalidArgument("reshape cannot infer the missing "
90 "Input to reshape is a SparseTensor with ", dense_size,
98 errors::InvalidArgument("Input to reshape is a tensor with ", dense_size,
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/layers/ |
| core.py | 319 @tf_export('keras.layers.Reshape')
320 class Reshape(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)
417 return K.reshape(inputs, (K.shape(inputs)[0],) + self.target_shape)
421 base_config = super(Reshape, self).get_config()
|
| /external/tensorflow/tensorflow/contrib/lite/testing/ |
| parse_testdata.cc | 293 // Process an 'reshape' message, triggering resizing of the input tensors via
294 // the test runner. A 'reshape' message looks like:
295 // reshape {
298 class Reshape : public Message {
300 explicit Reshape(TestRunner* test_runner) : test_runner_(test_runner) {
307 return test_runner_->Invalidate("Too many inputs to reshape");
338 } else if (s == "reshape") {
339 return Store(new Reshape(test_runner_));
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| operator.cc | 456 class Reshape
806 ops.emplace_back(new Reshape(::tflite::BuiltinOperator_RESHAPE,
[all...] |
| /external/tensorflow/tensorflow/contrib/specs/python/ |
| specs_ops.py | 111 Reshape = Fun(array_ops.reshape)
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| local_computation_builder.cc | 345 ComputationDataHandle LocalComputationBuilder::Reshape(
349 return builder_.Reshape(operand, dimensions, new_sizes);
|
| xla_client.py | 668 def Reshape(self, operand, dimensions, new_sizes):
669 """Reshape op."""
671 self._client.Reshape(
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| literal_test_util.cc | 756 /* static */ std::unique_ptr<Literal> LiteralTestUtil::Reshape(
|
| /frameworks/ml/nn/tools/test_generator/ |
| test_generator.py | 534 def Reshape(self, input, shape):
537 op = Operation("RESHAPE", ins, outs)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| computation_builder.cc | 377 ComputationDataHandle ComputationBuilder::Reshape(
393 ComputationDataHandle ComputationBuilder::Reshape(
406 return Reshape(operand, dimensions, new_sizes);
441 // enqueueing a trivial reshape.
457 return Reshape(operand, new_sizes);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| literal_util.cc | 736 StatusOr<std::unique_ptr<Literal>> Literal::Reshape(
739 return InvalidArgument("Reshape does not support tuples.");
756 "Shapes before and after Literal::Reshape have different numbers "
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
