Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dtranspose_conv_test.cc111 int output_shape_; member in tflite::__anonf3f7b7120111::BaseTransposeConvOpModel
671 int output_shape_; member in tflite::__anonf3f7b7120111::BaseTransposeConvBiasOpModel
H A Dsparse_to_dense_test.cc86 int output_shape_; member in tflite::__anon0fc1961c0111::SparseToDenseOpModel
H A Dconv3d_transpose_test.cc108 int output_shape_; member in tflite::__anon00f8d9890111::Conv3dTransposeOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dreshape_builder.h41 std::vector<int> output_shape_; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
H A Dconvolution_op_builder.h80 TfLiteTensor* output_shape_ = nullptr; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dfft_thunk.h88 const Shape output_shape_; variable
H A Dgpu_executable.h280 xla::Shape output_shape_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dxla_custom_call_op.cc63 TensorShape output_shape_; member in tensorflow::__anon73034a840111::XlaCustomCallOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dwinograd_transform.h52 const Shape output_shape_; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dconcatenation_tester.h87 std::vector<int32_t> output_shape_; variable
H A Dreshape_tester.h78 std::vector<int32_t> output_shape_; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/
H A Dadd.cc100 std::vector<int32_t> output_shape_; member in tflite::__anond1ed37de0111::AddModelBuilder
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Ddepthwiseconv_multithread.h73 const RuntimeShape& output_shape_; member
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dtranspose_conv_test.cc107 int output_shape_; member in tflite::QuantizedTransposeConvOpModel
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/
H A Dhost_compute_ops.cc520 TensorShape output_shape_; member in tensorflow::__anonf8e348410111::RecvFromHostOp
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
H A Ddepthwise_conv_plus_1x1_conv.cc91 BHWC output_shape_; member in tflite::gpu::ThinPointwiseFuser