/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/ |
D | reshape_builder.cc | 78 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()
|
D | reshape_builder.h | 41 std::vector<int> output_shape_; variable
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | fft_thunk.cc | 107 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()
|
D | fft_thunk.h | 89 const Shape output_shape_; variable
|
/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/tests/ |
D | transpose_conv_test.cc | 32 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/ |
D | transpose_conv_test.cc | 58 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
|
D | sparse_to_dense_test.cc | 37 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/ |
D | matmul_bcast.h | 42 const TensorShape& output_batch_shape() const { return output_shape_; } in output_batch_shape() 62 TensorShape output_shape_; variable
|
D | matmul_bcast.cc | 29 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/ |
D | depthwiseconv_multithread.h | 49 output_shape_(output_shape), in DepthwiseConvWorkerTask() 58 filter_data_, bias_shape_, bias_data_, output_shape_, in Run() 71 const RuntimeShape& output_shape_; member
|
D | legacy_optimized_ops.h | 396 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/ |
D | winograd_transform.h | 34 : 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/ |
D | nn_ops.py | 2364 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/ |
D | depthwise_conv.h | 1872 output_shape_(output_shape), 1881 bias_data_, output_shape_, output_data_, thread_start_, 1895 const RuntimeShape& output_shape_;
|