Searched refs:sizes_value (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/proto/ |
| D | encode_proto_op_test_base.py | 86 for sizes_value in 1, np.array([[[0, 0]]]):
92 sizes=sizes_value,
105 sizes: sizes_value,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2217 const Tensor* sizes_value, in SliceHelper() argument
2219 auto sizes_vec = sizes_value->vec<T>(); in SliceHelper()
2220 for (int i = 0; i < sizes_value->NumElements(); ++i) { in SliceHelper()
2266 const Tensor* sizes_value = c->input_tensor(2); in SliceShape() local
2268 if (sizes_value != nullptr) { in SliceShape()
2270 c->WithRank(begin_value, sizes_value->NumElements(), &begin_value)); in SliceShape()
2274 if (sizes_value->dtype() == DT_INT64) { in SliceShape()
2276 SliceHelper<int64_t>(c, begin_value, sizes_value, &dims)); in SliceShape()
2279 SliceHelper<int32>(c, begin_value, sizes_value, &dims)); in SliceShape()
2293 ShapeHandle sizes_value; in SliceShape() local
[all …]
|