Home
last modified time | relevance | path

Searched defs:Reshape (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dquantized_reshape_op_test.cc43 TEST_F(QuantizedReshapeTest, Reshape) { in TEST_F() argument
Dreshape_util.cc34 void Reshape(OpKernelContext *context, const Tensor &input_indices_in, in Reshape() function
Dscoped_allocator_ops_test.cc218 TEST_F(ScopedAllocatorConcatOpTest, Reshape) { in TEST_F() argument
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dparse_testdata.cc299 class Reshape : public Message { class
301 explicit Reshape(TestRunner* test_runner) : test_runner_(test_runner) { in Reshape() function in tflite::testing::Reshape
/external/tensorflow/tensorflow/lite/testing/
Dparse_testdata.cc296 class Reshape : public Message { class
298 explicit Reshape(TestRunner* test_runner) : test_runner_(test_runner) { in Reshape() function in tflite::testing::Reshape
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Dreshape.py65 class Reshape(bijector.Bijector): class
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md1960 ## Reshape section in Operation Semantics
/external/tensorflow/tensorflow/contrib/specs/python/
Dspecs_ops.py111 Reshape = Fun(array_ops.reshape) variable
/external/tensorflow/tensorflow/python/keras/layers/
Dcore.py371 class Reshape(Layer): class
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc713 StatusOr<XlaOp> XlaBuilder::Reshape(const Shape& shape, const XlaOp& operand) { in Reshape() function in xla::XlaBuilder
890 XlaOp XlaBuilder::Reshape(const XlaOp& operand, in Reshape() function in xla::XlaBuilder
905 XlaOp XlaBuilder::Reshape(const XlaOp& operand, in Reshape() function in xla::XlaBuilder
2855 XlaOp Reshape(const XlaOp& operand, absl::Span<const int64> dimensions, in Reshape() function
2860 XlaOp Reshape(const XlaOp& operand, absl::Span<const int64> new_sizes) { in Reshape() function
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc487 LocalOp ComputationBuilder::Reshape(const LocalOp& operand, in Reshape() function in xla::swig::ComputationBuilder
Dxla_client.py1188 def Reshape(self, operand, dimensions, new_sizes): member in ComputationBuilder
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc648 StatusOr<Literal> LiteralBase::Reshape( in Reshape() function in xla::LiteralBase
/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc409 TEST_F(TensorReshapeTest, Reshape) { in TEST_F() argument
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc857 class Reshape class
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2661 TEST_F(OpTest, Reshape) { in TEST_F() argument
/external/tensorflow/tensorflow/go/op/
Dwrappers.go2113 func Reshape(scope *Scope, tensor tf.Output, shape tf.Output) (output tf.Output) { func