Searched refs:input_shape_1 (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_input_conversion_op.cc | 77 MklDnnShape input_shape_1; in Compute() local
78 GetMklShape(context, kInputIndex_1, &input_shape_1); 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()
102 TensorShape tf_shape1 = input_shape_1.GetTfShape(); in Compute()
107 auto input1_md = input_shape_1.GetMklLayout(); in Compute()
159 bool mkl_shapes_are_same = ((input_shape_0 == input_shape_1) && in Compute()
194 if (input_shape_0.IsMklTensor() && !input_shape_1.IsMklTensor()) { in Compute()
200 } else if (!input_shape_0.IsMklTensor() && input_shape_1.IsMklTensor()) { in Compute()
202 mkl_shape = &input_shape_1; in Compute()
|
| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | test_manifest.txt | 1 add/add_activation=True,dtype=tf.float32,input_shape_1=[1,3,4,3],input_shape_2=[1,3,4,3]
2 DISABLED_add/add_activation=True,dtype=tf.int32,input_shape_1=[1,3,4,3],input_shape_2=[1,3,4,3]
3 add/add_activation=False,dtype=tf.float32,input_shape_1=[5],input_shape_2=[5]
4 add/add_activation=True,dtype=tf.float32,input_shape_1=[5],input_shape_2=[5]
5 add/add_activation=True,dtype=tf.float32,input_shape_1=[1,3,4,3],input_shape_2=[3]
6 add/add_activation=False,dtype=tf.float32,input_shape_1=[1,3,4,3],input_shape_2=[3]
7 DISABLED_add/add_activation=True,dtype=tf.int32,input_shape_1=[1,3,4,3],input_shape_2=[3]
8 DISABLED_add/add_activation=False,dtype=tf.int32,input_shape_1=[1,3,4,3],input_shape_2=[3]
9 DISABLED_add/add_activation=True,dtype=tf.int64,input_shape_1=[1,3,4,3],input_shape_2=[3]
10 DISABLED_add/add_activation=False,dtype=tf.int64,input_shape_1=[1,3,4,3],input_shape_2=[3]
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1315 const MklShape* input_shape_1) { in MklCompareShapes() argument
1317 if (input_shape_0->GetDimension() != input_shape_1->GetDimension()) { in MklCompareShapes()
1324 if (input_shape_0->dim_size(i) != input_shape_1->dim_size(i)) { in MklCompareShapes()
1335 const TensorShape* input_shape_1) { in MklCompareShapes() argument
1337 if (input_shape_0->GetDimension() != input_shape_1->dims()) { in MklCompareShapes()
1344 if (input_shape_0->tf_dim_size(i) != input_shape_1->dim_size(i)) { in MklCompareShapes()
1355 const MklShape* input_shape_1) { in MklCompareShapes() argument
1356 return MklCompareShapes(input_shape_1, input_shape_0); in MklCompareShapes()
1362 const TensorShape* input_shape_1) { in MklCompareShapes() argument
1364 if (input_shape_0->dims() != input_shape_1->dims()) { in MklCompareShapes()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 152 int64 CwiseOutputElementCount(const TensorShapeProto& input_shape_1, in CwiseOutputElementCount() argument
155 int rank = std::max(1, input_shape_1.dim_size()); in CwiseOutputElementCount()
157 MaybeGetMinimumShape(input_shape_1, rank, &found_unknown_shapes); in CwiseOutputElementCount()
159 if (input_shape_1.dim_size() == input_shape_2.dim_size()) { in CwiseOutputElementCount()
161 MaybeGetMinimumShape(input_shape_1, rank, &found_unknown_shapes); in CwiseOutputElementCount()
|