Home
last modified time | relevance | path

Searched refs:shape_and_slice (Results 1 – 17 of 17) 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/
Dio_ops.cc110 const string& shape_and_slice = shape_and_slices_flat(i); in __anone43acdb40302() local
111 if (shape_and_slice.empty()) { in __anone43acdb40302()
119 shape_and_slice, &parsed_full_shape, &parsed_slice, in __anone43acdb40302()
224 const auto& shape_and_slice = in __anone43acdb40802() local
226 if (shape_and_slice.empty()) { in __anone43acdb40802()
233 shape_and_slice, &parsed_full_shape, &parsed_slice, in __anone43acdb40802()
Dops.pbtxt31336 name: "shape_and_slice"
/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_tensor.cc276 if (shape_and_slice.empty()) { in run()
288 checkpoint::ParseShapeAndSlice(shape_and_slice, &parsed_full_shape, in run()
309 string shape_and_slice; member
362 const string& shape_and_slice = shape_and_slices_flat(i); in RestoreTensorsV2() local
364 new RestoreOp{context, i, tensor_name, shape_and_slice, prefix_string}; in RestoreTensorsV2()
Drestore_op_test.cc433 string shape_and_slice = "4 16 0,2:-"; in TEST_F() local
441 AddInput<string>(TensorShape({}), [&shape_and_slice](int x) -> string { in TEST_F()
442 return shape_and_slice; in TEST_F()
/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.cc158 const string& shape_and_slice, Tensor* tensor) { in ReadTensorFromCheckpoint() argument
165 if (!shape_and_slice.empty()) { in ReadTensorFromCheckpoint()
167 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.cc829 Output shape_and_slice = ops::Const(s.WithOpName("shape_and_slice"), in TEST_F() local
833 shape_and_slice, DataType::DT_FLOAT); in TEST_F()
839 shape_and_slice, DataType::DT_FLOAT); in TEST_F()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt3033 …argspec: "args=[\'file_pattern\', \'tensor_name\', \'shape_and_slice\', \'dt\', \'preferred_shard\…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt3033 …argspec: "args=[\'file_pattern\', \'tensor_name\', \'shape_and_slice\', \'dt\', \'preferred_shard\…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go27731 func RestoreSlice(scope *Scope, file_pattern tf.Output, tensor_name tf.Output, shape_and_slice tf.O…
27742 file_pattern, tensor_name, shape_and_slice,
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt27823 name: "shape_and_slice"
Dops_history.v2.pbtxt62899 name: "shape_and_slice"
62929 name: "shape_and_slice"
Dops_history.v1.pbtxt63526 name: "shape_and_slice"
63556 name: "shape_and_slice"