Searched defs:src_dims (Results 1 – 9 of 9) sorted by relevance
320 memory::dims src_dims(src_md.data.dims, in Compute() local333 memory::dims src_dims = TFShapeToMklDnnDims(input_tensors[k].shape()); in Compute() local507 auto src_dims = TFShapeToMklDnnDims(input_shapes[k].GetTfShape()); in FindMklCommonFormat() local
63 auto src_dims = TFShapeToMklDnnDims(src_tf_shape); in Compute() local
43 memory::dims src_dims; // check if this is needed member in tensorflow::MklEltwiseFwdParams49 MklEltwiseFwdParams(memory::dims src_dims, memory::desc src_md, in MklEltwiseFwdParams()226 memory::dims src_dims; member in tensorflow::MklEltwiseBwdParams232 MklEltwiseBwdParams(const memory::dims& src_dims, in MklEltwiseBwdParams()471 memory::dims src_dims; in Compute() local586 memory::dims src_dims = {}; in Compute() local
36 memory::dims src_dims; member272 memory::dims src_dims; member610 memory::dims src_dims = in Compute() local924 memory::dims src_dims = in Compute() local
483 memory::dims src_dims = in Compute() local
52 memory::dims src_dims; member
561 memory::dims src_dims = in Compute() local
67 memory::dims src_dims; member940 memory::dims src_dims, filter_dims, padding_left, padding_right, in Compute() local
46 memory::dims src_dims; member