Home
last modified time | relevance | path

Searched refs:shape_and_slice (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dsaved_tensor_slice_util.cc90 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()
Dsaved_tensor_slice_util.h56 Status ParseShapeAndSlice(const string& shape_and_slice, TensorShape* shape,
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_RestoreSlice.pbtxt18 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/core/ops/compat/ops_history_v2/
DRestoreSlice.pbtxt12 name: "shape_and_slice"
42 name: "shape_and_slice"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DRestoreSlice.pbtxt12 name: "shape_and_slice"
42 name: "shape_and_slice"
/external/tensorflow/tensorflow/core/ops/
Dio_ops.cc110 const string& shape_and_slice = shape_and_slices_flat(i); in __anon52e493120302() local
111 if (shape_and_slice.empty()) { in __anon52e493120302()
119 shape_and_slice, &parsed_full_shape, &parsed_slice, in __anon52e493120302()
224 const auto& shape_and_slice = in __anon52e493120802() local
226 if (shape_and_slice.empty()) { in __anon52e493120802()
233 shape_and_slice, &parsed_full_shape, &parsed_slice, in __anon52e493120802()
Dops.pbtxt40550 name: "shape_and_slice"
/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_tensor.cc277 if (shape_and_slice.empty()) { in run()
289 checkpoint::ParseShapeAndSlice(shape_and_slice, &parsed_full_shape, in run()
328 string shape_and_slice; member
381 const string& shape_and_slice = shape_and_slices_flat(i); in RestoreTensorsV2() local
383 new RestoreOp{context, i, tensor_name, shape_and_slice, prefix_string}; in RestoreTensorsV2()
Drestore_op_test.cc436 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/training/tracking/
Dbase.py107 shape_and_slice = full_shape_str + slice_spec
111 shape_and_slice = ""
113 {VARIABLE_VALUE_KEY: shape_and_slice})[VARIABLE_VALUE_KEY]
357 shape_and_slice = shape_and_slices[serialized_tensor.name]
359 shape_and_slice = ""
363 shape_and_slices=[shape_and_slice],
/external/tensorflow/tensorflow/python/ops/
Dio_ops.py75 def _restore_slice(file_pattern, tensor_name, shape_and_slice, tensor_type, argument
95 file_pattern, tensor_name, shape_and_slice, base_type,
/external/tensorflow/tensorflow/tools/graph_transforms/
Dsparsify_gather.cc157 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()
Dsparsify_gather_test.cc37 const string& shape_and_slice, Tensor* tensor);
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc867 Output shape_and_slice = ops::Const(s.WithOpName("shape_and_slice"), in TEST_F() local
871 shape_and_slice, DataType::DT_FLOAT); in TEST_F()
877 shape_and_slice, DataType::DT_FLOAT); in TEST_F()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt3733 …argspec: "args=[\'file_pattern\', \'tensor_name\', \'shape_and_slice\', \'dt\', \'preferred_shard\…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt3733 …argspec: "args=[\'file_pattern\', \'tensor_name\', \'shape_and_slice\', \'dt\', \'preferred_shard\…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go14476 func RestoreSlice(scope *Scope, file_pattern tf.Output, tensor_name tf.Output, shape_and_slice tf.O…
14487 file_pattern, tensor_name, shape_and_slice,
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt27823 name: "shape_and_slice"
Dops_history.v2.pbtxt62236 name: "shape_and_slice"
62266 name: "shape_and_slice"