Searched refs:dst_dims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_concat_op.cc | 279 memory::dims dst_dims; in Compute() local
282 dst_dims = TFShapeToMklDnnDims(mkl_input_shapes[0].GetTfShape()); in Compute()
287 dst_dims = TFShapeToMklDnnDims(input_tensors[0].shape()); in Compute()
346 dst_dims[concat_dim] = dst_concat_dim_size; in Compute()
356 dst_dims, MklDnnDataFormatToTFDataFormat(orig_tf_format)); in Compute()
364 dst_md = memory::desc(dst_dims, MklDnnType<T>(), memory::format::nchw); in Compute()
402 dnn_shape_dst.SetTfLayout(dst_dims.size(), dst_dims_in_nchw, in Compute()
407 tf_shape_dst = MklDnnDimsToTFShape(dst_dims); in Compute()
|
| D | mkl_pooling_ops_common.cc | 59 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 new memory::desc({bwdParams.dst_dims}, MklDnnType<T>(), in Setup()
149 get_desired_format(bwdParams.dst_dims[1], is_2d))); in Setup()
169 context_.diff_dst_fmt = get_desired_format(bwdParams.dst_dims[1], is_2d); in Setup()
175 {{{bwdParams.dst_dims}, MklDnnType<T>(), context_.diff_dst_fmt}, in Setup()
|
| D | mkl_pooling_ops_common.h | 47 memory::dims dst_dims; member
55 MklPoolingParams(memory::dims src_dims, memory::dims dst_dims, in MklPoolingParams()
60 dst_dims(dst_dims), in MklPoolingParams()
189 key_creator.AddAsKey(fwdParams.dst_dims); in CreateKey()
341 key_creator.AddAsKey(bwdParams.dst_dims); in CreateKey()
|
| D | mkl_conv_ops.cc | 70 memory::dims dst_dims; member
83 memory::dims bias_dims, memory::dims dst_dims, in MklConvFwdParams()
89 dst_dims(dst_dims), in MklConvFwdParams()
223 {convFwdDims.dst_dims}, MklDnnType<Toutput>(), memory::format::any)); in Setup()
368 key_creator.AddAsKey(convFwdDims.dst_dims); in CreateKey()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 210 int32* dst_dims = DimsData(); in ReplaceWith() local
211 std::memcpy(dst_dims, dims_data, dimensions_count * sizeof(int32)); in ReplaceWith()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 225 int32* dst_dims = DimsData(); in ReplaceWith() local
226 std::memcpy(dst_dims, dims_data, dimensions_count * sizeof(int32)); in ReplaceWith()
|