Home
last modified time | relevance | path

Searched defs:output_shape_data (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_concat_op_gpu.cu.cc46 GpuDeviceArrayStruct<int64_t> output_shape_data, in SparseConcatKernel()
87 GpuDeviceArrayStruct<int64_t> output_shape_data, in SparseConcatPermuteKernel()
H A Dsparse_to_dense_op.cc237 auto output_shape_data = in ComputeAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dconv3d_transpose_test.cc38 std::initializer_list<int> output_shape_data, const TensorData& filter, in Conv3dTransposeOpModel()
70 std::initializer_list<int> output_shape_data, const TensorData& filter, in Conv3dTransposeOpModel()
H A Dtranspose_conv_test.cc58 std::initializer_list<int> output_shape_data, in BaseTransposeConvOpModel()
583 std::initializer_list<int> output_shape_data, in BaseTransposeConvBiasOpModel()
H A Dsparse_to_dense_test.cc43 std::initializer_list<int> output_shape_data, in SparseToDenseOpModel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dtranspose_conv_test.cc27 QuantizedTransposeConvOpModel(std::initializer_list<int> output_shape_data, in QuantizedTransposeConvOpModel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc1322 std::vector<int>* output_shape_data = output_shape.mutable_dims(); in ProcessSpaceToBatchNDOperator() local
1375 std::vector<int>* output_shape_data = output_shape.mutable_dims(); in ProcessBatchToSpaceNDOperator() local
1894 const std::vector<int64_t>& output_shape_data = in ProcessSparseToDenseOperator() local