Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dreshape_test.cc30 TEST_F(OpenCLOperationTest, Reshape) { in TEST_F() argument
/external/tensorflow/tensorflow/core/kernels/
Dquantized_reshape_op_test.cc43 TEST_F(QuantizedReshapeTest, Reshape) { in TEST_F() argument
Dscoped_allocator_ops_test.cc219 TEST_F(ScopedAllocatorConcatOpTest, Reshape) { in TEST_F() argument
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dreshape.cc36 class Reshape : public NodeShader { class
Dreshape_test.cc98 TEST(Reshape, BatchIsUnsupported) { in TEST() 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.cc349 class Reshape : public Message { class
351 explicit Reshape(TestRunner* test_runner) : test_runner_(test_runner) {} in Reshape() function in tflite::testing::Reshape
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreshape_test.cc200 TEST(Reshape, MultiThreading) { in TEST() argument
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2326 ## Reshape section in Operation Semantics
/external/tensorflow/tensorflow/go/
Dtensor.go221 func (t *Tensor) Reshape(newShape []int64) error { func
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_normalization_test.cc111 TEST_F(LayoutNormalizationTest, Reshape) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Dlayer_utils.h626 StatusOr<nvinfer1::IShuffleLayer*> Reshape(nvinfer1::ITensor* input, in Reshape() function
/external/tensorflow/tensorflow/python/keras/layers/
Dcore.py444 class Reshape(Layer): class
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1182 XlaOp XlaBuilder::Reshape(XlaOp operand, absl::Span<const int64_t> dimensions, in Reshape() function in xla::XlaBuilder
1197 XlaOp XlaBuilder::Reshape(XlaOp operand, absl::Span<const int64_t> new_sizes, in Reshape() function in xla::XlaBuilder
1207 XlaOp XlaBuilder::Reshape(const Shape& shape, XlaOp operand, in Reshape() function in xla::XlaBuilder
4080 XlaOp Reshape(const XlaOp operand, absl::Span<const int64_t> dimensions, in Reshape() function
4085 XlaOp Reshape(const XlaOp operand, absl::Span<const int64_t> new_sizes) { in Reshape() function
4089 XlaOp Reshape(const Shape& shape, XlaOp operand) { in Reshape() function
/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc439 TEST_F(TensorReshapeTest, Reshape) { in TEST_F() argument
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc720 class Reshape class
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc958 StatusOr<Literal> LiteralBase::Reshape( in Reshape() function in xla::LiteralBase
/external/ComputeLibrary/
Dfilelist.json696 "Reshape": { object
2021 "Reshape": { object
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3837 TEST_F(OpTest, Reshape) { in TEST_F() argument
/external/tensorflow/tensorflow/go/op/
Dwrappers.go36725 func Reshape(scope *Scope, tensor tf.Output, shape tf.Output) (output tf.Output) { func