Searched refs:sizes_value (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/proto/python/kernel_tests/ |
| D | encode_proto_op_test_base.py | 90 for sizes_value in 1, np.array([[[0, 0]]]):
96 sizes=sizes_value,
109 sizes: sizes_value,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1393 const Tensor* sizes_value, in SliceHelper() argument
1395 auto sizes_vec = sizes_value->vec<T>(); in SliceHelper()
1396 for (int i = 0; i < sizes_value->NumElements(); ++i) { in SliceHelper()
1442 const Tensor* sizes_value = c->input_tensor(2); in SliceShape() local
1444 if (sizes_value != nullptr) { in SliceShape()
1446 c->WithRank(begin_value, sizes_value->NumElements(), &begin_value)); in SliceShape()
1450 if (sizes_value->dtype() == DT_INT64) { in SliceShape()
1452 SliceHelper<int64>(c, begin_value, sizes_value, &dims)); in SliceShape()
1455 SliceHelper<int32>(c, begin_value, sizes_value, &dims)); in SliceShape()
1469 ShapeHandle sizes_value; in SliceShape() local
[all …]
|