Searched defs:shape_and_slice (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | io_ops.cc | 115 const string& shape_and_slice = shape_and_slices_flat(i); in __anon2a9f7d410302() local
230 const auto& shape_and_slice = in __anon2a9f7d410802() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | saved_tensor_slice_util.cc | 90 Status ParseShapeAndSlice(const string& shape_and_slice, TensorShape* shape, in ParseShapeAndSlice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | save_restore_util.cc | 70 std::string shape_and_slice = absl::StrCat(shape_spec, " ", slice_spec); in BuildSliceSpecDeviceMap() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | io_ops.py | 72 def _restore_slice(file_pattern, tensor_name, shape_and_slice, tensor_type, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | save_restore_tensor.cc | 355 string shape_and_slice; member
|
| H A D | restore_op_test.cc | 436 string shape_and_slice = "4 16 0,2:-"; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | sparsify_gather.cc | 157 const string& shape_and_slice, Tensor* tensor) { in ReadTensorFromCheckpoint()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | save_restore_spmd_expander.cc | 206 for (const std::string& shape_and_slice : original_shape_and_slices) { in ConditionalSave() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | graph_properties_test.cc | 1035 Output shape_and_slice = ops::Const(s.WithOpName("shape_and_slice"), in TEST_F() local
|