Searched refs:SliceTensorShape (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_set.cc | 35 TF_RETURN_IF_ERROR(slice.SliceTensorShape(shape_, &result_shape)); in Register()
80 s = slice.SliceTensorShape(shape_, &target_shape); in QueryMeta()
92 s = intersection.SliceTensorShape(shape_, &inter_shape); in QueryMeta()
|
| D | tensor_slice_util.h | 155 s = slice_s.SliceTensorShape(shape, &shp_s);
160 s = slice_d.SliceTensorShape(shape, &shp_d);
|
| D | saved_tensor_slice_util.cc | 122 return slice->SliceTensorShape(*shape, shape_slice); in ParseShapeAndSlice()
|
| D | tensor_slice_reader.h | 186 Status s = slice_s.SliceTensorShape(tss->shape(), &shp_s); in CopySliceData()
|
| D | tensor_slice_writer.h | 149 TF_RETURN_IF_ERROR(slice.SliceTensorShape(saved_shape, &sliced_shape)); in Add()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_slice_test.cc | 234 TEST(TensorSliceTest, SliceTensorShape) { in TEST() argument
240 TF_EXPECT_OK(a.SliceTensorShape(x, &y)); in TEST()
249 Status s = a.SliceTensorShape(x, &y); in TEST()
|
| D | tensor_slice.h | 154 Status SliceTensorShape(const TensorShape& shape,
|
| D | tensor_slice.cc | 270 Status TensorSlice::SliceTensorShape(const TensorShape& shape, in SliceTensorShape() function in tensorflow::TensorSlice
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | load_and_remap_matrix_op.cc | 216 tensor_slice.SliceTensorShape(tensor_shape, &slice_shape)); in Compute()
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 364 slice_spec.SliceTensorShape(full_tensor_shape, &sliced_shape).IgnoreError(); in IsFullSlice()
|