Searched refs:dst_dims (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 240 memory::dims dst_dims; member
246 memory::dims dst_dims, int num_inputs, int concat_dims, in MklConcatFwdParams()
248 : dst_dims(dst_dims), in MklConcatFwdParams()
353 context_.dst_md.reset(new memory::desc({concat_fwd_dims.dst_dims}, in Setup()
580 memory::dims dst_dims; in Compute() local
583 dst_dims = TFShapeToMklDnnDims(mkl_input_shapes[0].GetTfShape()); in Compute()
588 dst_dims = TFShapeToMklDnnDims(input_tensors[0].shape()); in Compute()
661 dst_dims[concat_dim] = dst_concat_dim_size; in Compute()
671 if (dst_dims.size() == 4) { in Compute()
673 dst_dims, MklDnnDataFormatToTFDataFormat(orig_tf_format)); in Compute()
[all …]
|
| D | mkl_matmul_op_fused.cc | 111 memory::dims dst_dims = memory::dims({batch, channel}); in Compute() local
120 src_dims, weight_dims, bias_dims, dst_dims, src_format, in Compute()
167 : memory::desc(dst_dims, MklDnnType<T>(), output_format_tag); in Compute()
169 memory::desc(dst_dims, MklDnnType<T>(), output_format_tag); in Compute()
|
| D | mkl_pooling_ops_common.h | 42 memory::dims dst_dims; member
53 MklPoolingParams(memory::dims src_dims, memory::dims dst_dims, in MklPoolingParams()
60 dst_dims(dst_dims), in MklPoolingParams()
190 key_creator.AddAsKey(fwdParams.dst_dims); in CreateKey()
335 key_creator.AddAsKey(bwdParams.dst_dims); in CreateKey()
|
| D | mkl_matmul_ops_common.h | 50 memory::dims dst_dims; member
62 memory::dims bias_dims, memory::dims dst_dims,
69 dst_dims(dst_dims), in src_dims()
187 context_.dst_md.reset(new memory::desc({matmul_fwd_params.dst_dims}, in Setup()
339 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()
141 context_.dst_md.reset(new memory::desc({bwdParams.dst_dims}, MklDnnType<T>(), in Setup()
|
| D | mkl_conv_ops.cc | 62 memory::dims dst_dims; member
79 memory::dims bias_dims, memory::dims dst_dims, in MklConvFwdParams()
86 dst_dims(dst_dims), in MklConvFwdParams()
225 {convFwdDims.dst_dims}, MklDnnType<Toutput>(), user_data_fmt)); in Setup()
368 key_creator.AddAsKey(convFwdDims.dst_dims); in CreateKey()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 257 int32_t* dst_dims = DimsData(); in ReplaceWith() local
258 std::memcpy(dst_dims, dims_data, dimensions_count * sizeof(int32_t)); in ReplaceWith()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 420 int32_t* dst_dims = DimsData();
421 std::memcpy(dst_dims, dims_data, dimensions_count * sizeof(int32_t));
|