Home
last modified time | relevance | path

Searched refs:output_shape_ (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/
Dreshape_builder.cc78 PopulateOutputShapeFromTensor(shape_tensor, &output_shape_); in PopulateSubGraph()
82 PopulateShapeFromParam(reshape_params, &output_shape_); in PopulateSubGraph()
84 int num_elements_in_shape = static_cast<int>(output_shape_.size()); in PopulateSubGraph()
88 reinterpret_cast<char*>(output_shape_.data()), in PopulateSubGraph()
Dreshape_builder.h41 std::vector<int> output_shape_; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.cc107 output_shape_(output_shape) {} in FftThunk()
116 << ShapeUtil::HumanStringWithLayout(output_shape_); in ExecuteOnStream()
143 output_embed[i] = output_shape_.dimensions(dim_offset); in ExecuteOnStream()
144 output_distance *= output_shape_.dimensions(dim_offset); in ExecuteOnStream()
178 .ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
202 .ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
Dfft_thunk.h89 const Shape output_shape_; variable
/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/tests/
Dtranspose_conv_test.cc32 output_shape_ = AddConstInput(TensorType_INT32, output_shape_data, {4}); in QuantizedTransposeConvOpModel()
43 {GetShape(output_shape_), GetShape(filter_), GetShape(input_)}); in QuantizedTransposeConvOpModel()
58 int output_shape_; member in tflite::QuantizedTransposeConvOpModel
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_conv_test.cc58 output_shape_ = AddInput({TensorType_INT32, {4}}); in BaseTransposeConvOpModel()
61 output_shape_ = AddConstInput(TensorType_INT32, output_shape_data, {4}); in BaseTransposeConvOpModel()
75 {GetShape(output_shape_), GetShape(filter_), GetShape(input_)}); in BaseTransposeConvOpModel()
78 PopulateTensor<int32_t>(output_shape_, output_shape_data); in BaseTransposeConvOpModel()
96 int output_shape_; member in tflite::__anonf1ef55a70111::BaseTransposeConvOpModel
Dsparse_to_dense_test.cc37 output_shape_ = AddInput(TensorType_INT32); in SparseToDenseOpModel()
51 int output_shape() { return output_shape_; } in output_shape()
59 int output_shape_; member in tflite::__anonccad07510111::SparseToDenseOpModel
/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast.h42 const TensorShape& output_batch_shape() const { return output_shape_; } in output_batch_shape()
62 TensorShape output_shape_; variable
Dmatmul_bcast.cc29 output_shape_ = TensorShape(batch_bcast_->output_shape()); in MatMulBCast()
30 output_batch_size_ = output_shape_.num_elements(); in MatMulBCast()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h49 output_shape_(output_shape), in DepthwiseConvWorkerTask()
58 filter_data_, bias_shape_, bias_data_, output_shape_, in Run()
71 const RuntimeShape& output_shape_; member
Dlegacy_optimized_ops.h396 output_shape_(output_shape), in LegacyDepthwiseConvWorkerTask()
405 output_shape_, output_data_, thread_start_, in Run()
417 const RuntimeShape& output_shape_; member
508 output_shape_(output_shape), in LegacyPerChannelDepthwiseConvWorkerTask()
517 filter_shape_, filter_data_, bias_shape_, bias_data_, output_shape_, in Run()
531 const RuntimeShape& output_shape_; member
1010 output_shape_(output_shape), in LegacyFullyConnectedAsGEMVWorkerTask()
1020 output_activation_max_, output_shape_, output_data_, row_start_, in Run()
1037 const RuntimeShape& output_shape_; member
2021 output_shape_(output_shape), in LegacyInt8FullyConnectedAsGEMVWorkerTask()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dwinograd_transform.h34 : filter_shape_(3, 3), input_shape_(4, 4), output_shape_(2, 2) {} in WinogradTransform()
47 virtual const Shape& output_shape() const { return output_shape_; } in output_shape()
52 const Shape output_shape_; variable
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py2364 output_shape_ = ops.convert_to_tensor(output_shape, name="output_shape")
2365 if not output_shape_.get_shape().is_compatible_with(
2368 output_shape_.get_shape()))
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv.h1872 output_shape_(output_shape),
1881 bias_data_, output_shape_, output_data_, thread_start_,
1895 const RuntimeShape& output_shape_;