Home
last modified time | relevance | path

Searched defs:output_shape_ (Results 1 – 17 of 17) 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.h80 TfLiteTensor* output_shape_ = nullptr; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.h88 const Shape output_shape_; variable
Dgpu_executable.h280 xla::Shape output_shape_; variable
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_custom_call_op.cc63 TensorShape output_shape_; member in tensorflow::__anon911e25040111::XlaCustomCallOp
/external/tensorflow/tensorflow/core/kernels/
Dwinograd_transform.h52 const Shape output_shape_; variable
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dconcatenation_tester.h87 std::vector<int32_t> output_shape_; variable
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::__anon51f9a25e0111::AddModelBuilder
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_conv_test.cc111 int output_shape_; member in tflite::__anond94771920111::BaseTransposeConvOpModel
671 int output_shape_; member in tflite::__anond94771920111::BaseTransposeConvBiasOpModel
Dsparse_to_dense_test.cc86 int output_shape_; member in tflite::__anon9f08a09c0111::SparseToDenseOpModel
Dconv3d_transpose_test.cc108 int output_shape_; member in tflite::__anon792134090111::Conv3dTransposeOpModel
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h73 const RuntimeShape& output_shape_; member
Dlegacy_optimized_ops.h417 const RuntimeShape& output_shape_; member
532 const RuntimeShape& output_shape_; member
1039 const RuntimeShape& output_shape_; member
2050 const RuntimeShape& output_shape_; member
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dtranspose_conv_test.cc107 int output_shape_; member in tflite::QuantizedTransposeConvOpModel
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc520 TensorShape output_shape_; member in tensorflow::__anon91ab92c10111::RecvFromHostOp
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Ddepthwise_conv_plus_1x1_conv.cc91 BHWC output_shape_; member in tflite::gpu::ThinPointwiseFuser