| /external/tensorflow/tensorflow/lite/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 | 115 output_shape_(output_shape) {} in FftThunk()
124 << ShapeUtil::HumanStringWithLayout(output_shape_); in ExecuteOnStream()
165 output_embed[i] = output_shape_.dimensions(dim_offset); in ExecuteOnStream()
166 output_distance *= output_shape_.dimensions(dim_offset); in ExecuteOnStream()
206 .ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
221 .ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
251 .ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
265 .ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
|
| D | fft_thunk.h | 97 const Shape output_shape_; variable
|
| D | gpu_executable.cc | 66 output_shape_(params.output_shape), in GpuExecutable()
438 ExecutionOutput result(/*on_device_shape=*/output_shape_, memory_allocator, in ExecuteAsyncOnStreamImpl()
519 ShapeUtil::ByteSizeOf(ShapeUtil::GetSubshape(output_shape_, index)); in ExecuteAsyncOnStreamImpl()
|
| D | gpu_executable.h | 198 xla::Shape output_shape_; variable
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | reshape_tester.h | 49 output_shape_ = in OutputShape()
56 return output_shape_; in OutputShape()
78 std::vector<int32_t> output_shape_; variable
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv_test.cc | 68 output_shape_ = AddInput({TensorType_INT32, {4}}); in BaseTransposeConvOpModel()
71 output_shape_ = AddConstInput(TensorType_INT32, output_shape_data, {4}); in BaseTransposeConvOpModel()
85 {GetShape(output_shape_), GetShape(filter_), GetShape(input_)}); in BaseTransposeConvOpModel()
88 PopulateTensor<int32_t>(output_shape_, output_shape_data); in BaseTransposeConvOpModel()
111 int output_shape_; member in tflite::__anon7fbb49030111::BaseTransposeConvOpModel
582 output_shape_ = AddInput({TensorType_INT32, {4}}); in BaseTransposeConvBiasOpModel()
585 output_shape_ = AddConstInput(TensorType_INT32, output_shape_data, {4}); in BaseTransposeConvBiasOpModel()
630 BuildInterpreter({GetShape(output_shape_), GetShape(filter_), in BaseTransposeConvBiasOpModel()
634 PopulateTensor<int32_t>(output_shape_, output_shape_data); in BaseTransposeConvBiasOpModel()
662 int output_shape_; member in tflite::__anon7fbb49030111::BaseTransposeConvBiasOpModel
|
| D | sparse_to_dense_test.cc | 38 output_shape_ = AddInput(TensorType_INT32); in SparseToDenseOpModel()
52 int output_shape() { return output_shape_; } in output_shape()
60 int output_shape_; member in tflite::__anon13f7662d0111::SparseToDenseOpModel
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/ |
| D | add.cc | 52 builder, builder.CreateVector<int32_t>(output_shape_.data(), 4), in Build()
95 output_shape_ = std::move(shape); in SetOutputShape()
100 std::vector<int32_t> output_shape_; member in tflite::__anon5ac0d3af0111::AddModelBuilder
|
| /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 | 395 output_shape_(output_shape), in LegacyDepthwiseConvWorkerTask()
404 output_shape_, output_data_, thread_start_, in Run()
416 const RuntimeShape& output_shape_; member
507 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
1011 output_shape_(output_shape), in LegacyFullyConnectedAsGEMVWorkerTask()
1021 output_activation_max_, output_shape_, output_data_, row_start_, in Run()
1038 const RuntimeShape& output_shape_; member
2022 output_shape_(output_shape), in LegacyInt8FullyConnectedAsGEMVWorkerTask()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | transpose_conv_test.cc | 35 output_shape_ = AddConstInput(TensorType_INT32, output_shape_data, {4}); in QuantizedTransposeConvOpModel()
79 {GetShape(output_shape_), GetShape(filter_), GetShape(input_)}); in QuantizedTransposeConvOpModel()
105 int output_shape_; member in tflite::QuantizedTransposeConvOpModel
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | convolution_op_builder.cc | 40 output_shape_ = output_shape; in SetOutputShape()
93 GetTensorData<int32_t>(output_shape_)[height_index]); in Build()
95 GetTensorData<int32_t>(output_shape_)[width_index]); in Build()
|
| D | convolution_op_builder.h | 78 TfLiteTensor* output_shape_ = nullptr; variable
|
| /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/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 447 OP_REQUIRES_OK(ctx, ctx->GetAttr("shape", &output_shape_)); in RecvFromHostOp()
471 ctx, TensorShapeToXLAShape(output_dtype_, output_shape_, &xla_shape)); in Compile()
490 TensorShape output_shape_; member in tensorflow::__anonc4b4b8d20111::RecvFromHostOp
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | depthwise_conv.h | 1913 output_shape_(output_shape),
1923 bias_data_, output_shape_, output_data_, thread_start_,
1937 const RuntimeShape& output_shape_;
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 2688 output_shape_ = ops.convert_to_tensor(output_shape, name="output_shape")
2689 if not output_shape_.get_shape().is_compatible_with(
2692 output_shape_.get_shape()))
|