| /external/tensorflow/tensorflow/core/util/ |
| D | saved_tensor_slice_util.cc | 90 Status ParseShapeAndSlice(const string& shape_and_slice, TensorShape* shape, in ParseShapeAndSlice() argument
92 CHECK(!shape_and_slice.empty()); in ParseShapeAndSlice()
95 std::vector<string> splits = str_util::Split(shape_and_slice, ' '); in ParseShapeAndSlice()
101 shape_and_slice); in ParseShapeAndSlice()
116 "Non numerical dimension in shape_and_slice: ", shape_and_slice); in ParseShapeAndSlice()
|
| D | saved_tensor_slice_util.h | 56 Status ParseShapeAndSlice(const string& shape_and_slice, TensorShape* shape,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_RestoreSlice.pbtxt | 18 name: "shape_and_slice"
46 only a slice of a larger tensor. `shape_and_slice` specifies the shape of the
49 The `shape_and_slice` input has the same format as the
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | save_restore_util.cc | 70 std::string shape_and_slice = absl::StrCat(shape_spec, " ", slice_spec); in BuildSliceSpecDeviceMap() local
74 if (min_device_for_slice_spec.find(shape_and_slice) == in BuildSliceSpecDeviceMap()
76 device_id < min_device_for_slice_spec[shape_and_slice]) { in BuildSliceSpecDeviceMap()
77 min_device_for_slice_spec[shape_and_slice] = device_id; in BuildSliceSpecDeviceMap()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | RestoreSlice.pbtxt | 12 name: "shape_and_slice"
42 name: "shape_and_slice"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | RestoreSlice.pbtxt | 12 name: "shape_and_slice"
42 name: "shape_and_slice"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | io_ops.cc | 115 const string& shape_and_slice = shape_and_slices_flat(i); in __anon711ce7c10302() local
116 if (shape_and_slice.empty()) { in __anon711ce7c10302()
124 shape_and_slice, &parsed_full_shape, &parsed_slice, in __anon711ce7c10302()
230 const auto& shape_and_slice = in __anon711ce7c10802() local
232 if (shape_and_slice.empty()) { in __anon711ce7c10802()
239 shape_and_slice, &parsed_full_shape, &parsed_slice, in __anon711ce7c10802()
|
| D | ops.pbtxt | 43962 name: "shape_and_slice"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_tensor.cc | 259 const string& shape_and_slice, const string& reader_prefix, in RestoreOp()
264 shape_and_slice(shape_and_slice), in RestoreOp()
304 if (shape_and_slice.empty()) { in run()
316 checkpoint::ParseShapeAndSlice(shape_and_slice, &parsed_full_shape, in run()
355 string shape_and_slice; member
|
| D | restore_op_test.cc | 436 string shape_and_slice = "4 16 0,2:-"; in TEST_F() local
444 AddInput<tstring>(TensorShape({}), [&shape_and_slice](int x) -> tstring { in TEST_F()
445 return shape_and_slice; in TEST_F()
|
| /external/tensorflow/tensorflow/python/trackable/ |
| D | base.py | 142 shape_and_slice = full_shape_str + slice_spec
144 shape_and_slice = ""
146 {VARIABLE_VALUE_KEY: shape_and_slice})[VARIABLE_VALUE_KEY]
|
| /external/tensorflow/tensorflow/python/checkpoint/ |
| D | restore.py | 126 shape_and_slice = shape_and_slices[serialized_tensor.name]
128 shape_and_slice = ""
132 shape_and_slices=[shape_and_slice],
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | io_ops.py | 72 def _restore_slice(file_pattern, tensor_name, shape_and_slice, tensor_type, argument
92 file_pattern, tensor_name, shape_and_slice, base_type,
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 157 const string& shape_and_slice, Tensor* tensor) { in ReadTensorFromCheckpoint() argument
164 if (!shape_and_slice.empty()) { in ReadTensorFromCheckpoint()
166 checkpoint::ParseShapeAndSlice(shape_and_slice, &parsed_full_shape, in ReadTensorFromCheckpoint()
|
| D | sparsify_gather_test.cc | 37 const string& shape_and_slice, Tensor* tensor);
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-113.md | 29 shape_and_slice='2',
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | save_restore_spmd_expander.cc | 206 for (const std::string& shape_and_slice : original_shape_and_slices) { in ConditionalSave() local
207 if (!shape_and_slice.empty()) in ConditionalSave()
211 shape_and_slice)); in ConditionalSave()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1035 Output shape_and_slice = ops::Const(s.WithOpName("shape_and_slice"), in TEST_F() local
1039 shape_and_slice, DataType::DT_FLOAT); in TEST_F()
1045 shape_and_slice, DataType::DT_FLOAT); in TEST_F()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3817 …argspec: "args=[\'file_pattern\', \'tensor_name\', \'shape_and_slice\', \'dt\', \'preferred_shard\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3817 …argspec: "args=[\'file_pattern\', \'tensor_name\', \'shape_and_slice\', \'dt\', \'preferred_shard\…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 39750 func RestoreSlice(scope *Scope, file_pattern tf.Output, tensor_name tf.Output, shape_and_slice tf.O…
39761 file_pattern, tensor_name, shape_and_slice,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 27823 name: "shape_and_slice"
|
| D | ops_history.v2.pbtxt | 62211 name: "shape_and_slice"
62241 name: "shape_and_slice"
|