Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc306 output_shape_(output_shape), in GemmThunk()
326 CHECK_EQ(dim_nums.lhs_batch_dimensions_size() + 2, output_shape_.rank()); in ExecuteOnStream()
330 int64 batch_size = std::accumulate(output_shape_.dimensions().begin(), in ExecuteOnStream()
331 output_shape_.dimensions().end() - 2, 1, in ExecuteOnStream()
342 for (const auto* shape : {&lhs_shape_, &rhs_shape_, &output_shape_}) { in ExecuteOnStream()
351 int64 output_num_rows = output_shape_.dimensions(row_dim); in ExecuteOnStream()
352 int64 output_num_cols = output_shape_.dimensions(col_dim); in ExecuteOnStream()
379 LayoutUtil::Minor(output_shape_.layout(), row_dim); in ExecuteOnStream()
396 PrimitiveType element_type = output_shape_.element_type(); in ExecuteOnStream()
423 ShapeUtil::ByteSizeOf(output_shape_)); in ExecuteOnStream()
[all …]
Dfft_thunk.cc107 output_shape_(output_shape) {} in FftThunk()
115 << ShapeUtil::HumanStringWithLayout(output_shape_); in ExecuteOnStream()
141 output_embed[i] = output_shape_.dimensions(dim_offset); in ExecuteOnStream()
142 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()
Dgemm_thunk.h71 const Shape output_shape_; variable
Dfft_thunk.h91 const Shape output_shape_; variable
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_conv_test.cc47 output_shape_ = AddInput({TensorType_INT32, {4}}); in TransposeConvOpModel()
60 {GetShape(output_shape_), GetShape(filter_), GetShape(input_)}); in TransposeConvOpModel()
64 PopulateTensor(output_shape_, i); in SetOutputShape()
74 int output_shape_; member in tflite::__anon558e5be50111::TransposeConvOpModel
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::__anona42cd54f0111::SparseToDenseOpModel
/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.py2283 output_shape_ = ops.convert_to_tensor(output_shape, name="output_shape")
2284 if not output_shape_.get_shape().is_compatible_with(tensor_shape.vector(4)):
2286 output_shape_.get_shape()))
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1058 output_shape_(output_shape), in FullyConnectedAsGEMVWorkerTask()
1068 output_activation_max_, output_shape_, output_data_, row_start_, in Run()
1085 const RuntimeShape& output_shape_; member
1889 output_shape_(output_shape), in MeanWorkerTask()
1898 input_scale_, output_shape_, output_data_, output_zero_point_, in Run()
1908 const RuntimeShape& output_shape_; member