Home
last modified time | relevance | path

Searched defs:output_shape_ (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dreshape_builder.h41 std::vector<int> output_shape_; variable
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dconvolution_op_builder.h78 TfLiteTensor* output_shape_ = nullptr; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.h97 const Shape output_shape_; variable
Dgpu_executable.h198 xla::Shape output_shape_; variable
/external/tensorflow/tensorflow/core/kernels/
Dwinograd_transform.h52 const Shape output_shape_; variable
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreshape_tester.h78 std::vector<int32_t> output_shape_; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/
Dadd.cc100 std::vector<int32_t> output_shape_; member in tflite::__anon5ac0d3af0111::AddModelBuilder
/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense_test.cc60 int output_shape_; member in tflite::__anon13f7662d0111::SparseToDenseOpModel
Dtranspose_conv_test.cc111 int output_shape_; member in tflite::__anon7fbb49030111::BaseTransposeConvOpModel
662 int output_shape_; member in tflite::__anon7fbb49030111::BaseTransposeConvBiasOpModel
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h71 const RuntimeShape& output_shape_; member
Dlegacy_optimized_ops.h416 const RuntimeShape& output_shape_; member
531 const RuntimeShape& output_shape_; member
1038 const RuntimeShape& output_shape_; member
2049 const RuntimeShape& output_shape_; member
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dtranspose_conv_test.cc105 int output_shape_; member in tflite::QuantizedTransposeConvOpModel
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc490 TensorShape output_shape_; member in tensorflow::__anonc4b4b8d20111::RecvFromHostOp