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 | 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 | 2126 const Tensor* sizes_value, in SliceHelper() argument
2128 auto sizes_vec = sizes_value->vec<T>(); in SliceHelper()
2129 for (int i = 0; i < sizes_value->NumElements(); ++i) { in SliceHelper()
2175 const Tensor* sizes_value = c->input_tensor(2); in SliceShape() local
2177 if (sizes_value != nullptr) { in SliceShape()
2179 c->WithRank(begin_value, sizes_value->NumElements(), &begin_value)); in SliceShape()
2183 if (sizes_value->dtype() == DT_INT64) { in SliceShape()
2185 SliceHelper<int64>(c, begin_value, sizes_value, &dims)); in SliceShape()
2188 SliceHelper<int32>(c, begin_value, sizes_value, &dims)); in SliceShape()
2202 ShapeHandle sizes_value; in SliceShape() local
[all …]
|