Home
last modified time | relevance | path

Searched defs:output_shape_ (Results 1 – 14 of 14) 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.h96 const Shape output_shape_; variable
Dgpu_executable.h204 xla::Shape output_shape_; variable
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreshape_tester.h78 std::vector<int32_t> output_shape_; variable
/external/tensorflow/tensorflow/core/kernels/
Dwinograd_transform.h52 const Shape 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::__anon085f3e940111::AddModelBuilder
/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense_test.cc60 int output_shape_; member in tflite::__anon893cca520111::SparseToDenseOpModel
Dtranspose_conv_test.cc111 int output_shape_; member in tflite::__anone8223c880111::BaseTransposeConvOpModel
671 int output_shape_; member in tflite::__anone8223c880111::BaseTransposeConvBiasOpModel
Dconv3d_transpose_test.cc108 int output_shape_; member in tflite::__anona9da947f0111::Conv3dTransposeOpModel
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h71 const RuntimeShape& output_shape_; member
Dlegacy_optimized_ops.h415 const RuntimeShape& output_shape_; member
530 const RuntimeShape& output_shape_; member
1037 const RuntimeShape& output_shape_; member
2048 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.cc507 TensorShape output_shape_; member in tensorflow::__anon68a9a7770111::RecvFromHostOp