| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/template_writer/cl/ |
| D | ClTemplateElementwiseBinary.cpp | 207 const auto &dst_dims = _dst->tensor_shape(); in get_tag_lut() local
209 const auto lhs_broadcast_x = dst_dims[0] != 1 && lhs_dims[0] == 1; in get_tag_lut()
210 const auto rhs_broadcast_x = dst_dims[0] != 1 && rhs_dims[0] == 1; in get_tag_lut()
211 const auto lhs_broadcast_y = dst_dims[1] != 1 && lhs_dims[1] == 1; in get_tag_lut()
212 const auto rhs_broadcast_y = dst_dims[1] != 1 && rhs_dims[1] == 1; in get_tag_lut()
213 const auto lhs_broadcast_z = dst_dims[2] != 1 && lhs_dims[2] == 1; in get_tag_lut()
214 const auto rhs_broadcast_z = dst_dims[2] != 1 && rhs_dims[2] == 1; in get_tag_lut()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 244 memory::dims dst_dims; member
250 memory::dims dst_dims, int num_inputs, int concat_dims, in MklConcatFwdParams()
252 : dst_dims(dst_dims), in MklConcatFwdParams()
360 context_.dst_md.reset(new memory::desc({concat_fwd_dims.dst_dims}, in Setup()
592 memory::dims dst_dims; in Compute() local
595 dst_dims = TFShapeToMklDnnDims(mkl_input_shapes[0].GetTfShape()); in Compute()
600 dst_dims = TFShapeToMklDnnDims(input_tensors[0].shape()); in Compute()
673 dst_dims[concat_dim] = dst_concat_dim_size; in Compute()
683 if (dst_dims.size() == 4) { in Compute()
685 dst_dims, MklDnnDataFormatToTFDataFormat(orig_tf_format)); in Compute()
[all …]
|
| D | mkl_matmul_op_fused.cc | 120 memory::dims dst_dims = memory::dims({batch, channel}); in Compute() local
127 src_dims, weight_dims, bias_dims, dst_dims, src_format, in Compute()
181 : memory::desc(dst_dims, MklDnnType<T>(), output_format_tag); in Compute()
183 memory::desc(dst_dims, MklDnnType<T>(), output_format_tag); in Compute()
|
| D | mkl_pooling_ops_common.h | 44 memory::dims dst_dims; member
55 MklPoolingParams(memory::dims src_dims, memory::dims dst_dims, in MklPoolingParams()
62 dst_dims(dst_dims), in MklPoolingParams()
196 key_creator.AddAsKey(fwdParams.dst_dims); in CreateKey()
344 key_creator.AddAsKey(bwdParams.dst_dims); in CreateKey()
|
| D | mkl_matmul_ops_common.h | 63 memory::dims dst_dims; member
80 memory::dims dst_dims,
88 dst_dims(dst_dims), in src_dims()
219 context_.dst_md.reset(new memory::desc({matmul_fwd_params.dst_dims}, in Setup()
405 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.dst_dims); in CreateKey()
|
| D | mkl_pooling_ops_common.cc | 47 context_.dst_md.reset(new memory::desc({fwdParams.dst_dims}, MklDnnType<T>(), in Setup()
144 context_.dst_md.reset(new memory::desc({bwdParams.dst_dims}, MklDnnType<T>(), in Setup()
|
| D | mkl_conv_ops.cc | 46 memory::dims dst_dims; member
67 memory::dims bias_dims, memory::dims dst_dims, in MklConvFwdParams()
75 dst_dims(dst_dims), in MklConvFwdParams()
292 {convFwdDims.dst_dims}, MklDnnType<Toutput>(), user_data_fmt)); in Setup()
492 key_creator.AddAsKey(convFwdDims.dst_dims); in CreateKey()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | runtime_shape.h | 122 int32_t* dst_dims = DimsData(); in ReplaceWith() local
123 std::memcpy(dst_dims, dims_data, dimensions_count * sizeof(int32_t)); in ReplaceWith()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_block_io.cpp | 195 auto dst_dims = blk_dims; in test_block_io_copy_using_reordered_dimensions() local
197 dst_dims[output_to_input_dim_map[out_dim]] = blk_dims[out_dim]; in test_block_io_copy_using_reordered_dimensions()
201 IODst dst(dst_dims, input_strides, output_data, first_coeff_index); in test_block_io_copy_using_reordered_dimensions()
|
| D | cxx11_tensor_block_eval.cpp | 143 DSizes<Index, NumDims> dst_dims; in VerifyBlockEvaluator() local
145 dst_dims = block_params.desc.dimensions(); in VerifyBlockEvaluator()
149 dst_dims[i] = block_params.desc.dimension(i) + extent; in VerifyBlockEvaluator()
154 Tensor<T, NumDims, Layout> dst(dst_dims); in VerifyBlockEvaluator()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBlock.h | 1153 Dst(const Dimensions& dst_dims, const Dimensions& dst_strides, Scalar* dst,
1155 : dims(dst_dims), strides(dst_strides), data(dst), offset(dst_offset) {}
|