Searched refs:input_shape_0 (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_input_conversion_op.cc | 73 MklDnnShape input_shape_0; in Compute() local
74 GetMklShape(context, kInputIndex_0, &input_shape_0); in Compute()
86 if (!input_shape_0.IsMklTensor() && !input_shape_1.IsMklTensor()) { in Compute()
97 if (input_shape_0.IsMklTensor() && input_shape_1.IsMklTensor()) { in Compute()
101 TensorShape tf_shape0 = input_shape_0.GetTfShape(); in Compute()
106 auto input0_md = input_shape_0.GetMklLayout(); in Compute()
127 mkl_output_mkl_shape.SetTfLayout(input_shape_0.GetDimension(), in Compute()
128 input_shape_0.GetSizesAsMklDnnDims(), in Compute()
129 input_shape_0.GetTfDataFormat()); in Compute()
159 bool mkl_shapes_are_same = ((input_shape_0 == input_shape_1) && in Compute()
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1314 inline bool MklCompareShapes(const MklShape* input_shape_0, in MklCompareShapes() argument
1317 if (input_shape_0->GetDimension() != input_shape_1->GetDimension()) { in MklCompareShapes()
1322 size_t ndims = input_shape_0->GetDimension(); in MklCompareShapes()
1324 if (input_shape_0->dim_size(i) != input_shape_1->dim_size(i)) { in MklCompareShapes()
1334 inline bool MklCompareShapes(const MklShape* input_shape_0, in MklCompareShapes() argument
1337 if (input_shape_0->GetDimension() != input_shape_1->dims()) { in MklCompareShapes()
1342 size_t ndims = input_shape_0->GetDimension(); in MklCompareShapes()
1344 if (input_shape_0->tf_dim_size(i) != input_shape_1->dim_size(i)) { in MklCompareShapes()
1354 inline bool MklCompareShapes(const TensorShape* input_shape_0, in MklCompareShapes() argument
1356 return MklCompareShapes(input_shape_1, input_shape_0); in MklCompareShapes()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 586 input_shape_0 = math_ops.cast(input_shape[0], grad.indices.dtype)
589 math_ops.mod(grad.indices, input_shape_0),
590 input_shape_0)
|