Searched refs:dst_dims (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| 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()
362 context_.dst_md.reset(new memory::desc({concat_fwd_dims.dst_dims}, in Setup()
579 memory::dims dst_dims; in Compute() local
582 dst_dims = TFShapeToMklDnnDims(mkl_input_shapes[0].GetTfShape()); in Compute()
587 dst_dims = TFShapeToMklDnnDims(input_tensors[0].shape()); in Compute()
660 dst_dims[concat_dim] = dst_concat_dim_size; in Compute()
669 if (dst_dims.size() == 4) { in Compute()
671 dst_dims, MklDnnDataFormatToTFDataFormat(orig_tf_format)); in Compute()
[all …]
|
| D | mkl_matmul_ops_common.h | 42 memory::dims dst_dims; member
52 memory::dims bias_dims, memory::dims dst_dims,
57 dst_dims(dst_dims), in src_dims()
168 context_.dst_md.reset(new memory::desc({matmul_fwd_params.dst_dims}, in Setup()
310 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.dst_dims); in CreateKey()
|
| D | mkl_matmul_op_fused.cc | 105 memory::dims dst_dims = memory::dims({batch, channel}); in Compute() local
110 dst_dims, weight_format); in Compute()
123 this->AllocateOutputTensor(ctx, *matmul_pd, dst_dims, memory::format::nc, in Compute()
|
| D | mkl_pooling_ops_common.cc | 58 context_.dst_md.reset(new memory::desc({fwdParams.dst_dims}, MklDnnType<T>(), in Setup()
143 bool is_2d = (bwdParams.dst_dims.size() == 4); in Setup()
148 {bwdParams.dst_dims}, MklDnnType<T>(), bwdParams.src_format)); in Setup()
174 {{{bwdParams.dst_dims}, MklDnnType<T>(), context_.diff_dst_fmt}, in Setup()
|
| D | mkl_pooling_ops_common.h | 43 memory::dims dst_dims; member
52 MklPoolingParams(memory::dims src_dims, memory::dims dst_dims, in MklPoolingParams()
58 dst_dims(dst_dims), in MklPoolingParams()
188 key_creator.AddAsKey(fwdParams.dst_dims); in CreateKey()
340 key_creator.AddAsKey(bwdParams.dst_dims); in CreateKey()
|
| D | mkl_conv_ops.cc | 65 memory::dims dst_dims; member
80 memory::dims bias_dims, memory::dims dst_dims, in MklConvFwdParams()
86 dst_dims(dst_dims), in MklConvFwdParams()
230 {convFwdDims.dst_dims}, MklDnnType<Toutput>(), MEMORY_FORMAT::any)); in Setup()
400 key_creator.AddAsKey(convFwdDims.dst_dims); in CreateKey()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 233 int32* dst_dims = DimsData(); in ReplaceWith() local
234 std::memcpy(dst_dims, dims_data, dimensions_count * sizeof(int32)); in ReplaceWith()
|