Home
last modified time | relevance | path

Searched refs:LookupSlice (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle_test.cc384 TF_ASSERT_OK(reader.LookupSlice("foo", distinct_slice, &val)); in TEST()
394 TF_ASSERT_OK(reader.LookupSlice("foo", distinct_slice, &val)); in TEST()
418 TF_ASSERT_OK(reader.LookupSlice("no_extents", slice, &val)); in TEST()
427 TF_ASSERT_OK(reader.LookupSlice("both_extents", slice, &val)); in TEST()
436 TF_ASSERT_OK(reader.LookupSlice("no_extents", slice, &val)); in TEST()
445 TF_ASSERT_OK(reader.LookupSlice("both_extents", slice, &val)); in TEST()
Dtensor_bundle.h243 Status LookupSlice(StringPiece full_tensor_key, const TensorSlice& slice_spec,
Dtensor_bundle.cc923 Status BundleReader::LookupSlice(StringPiece full_tensor_key, in LookupSlice() function in tensorflow::BundleReader
/external/tensorflow/tensorflow/core/kernels/
Dload_and_remap_matrix_op.cc213 OP_REQUIRES_OK(context, reader.LookupSlice(old_tensor_name, tensor_slice, in Compute()
Dsave_restore_tensor.cc301 reader->LookupSlice(tensor_name, parsed_slice, restored_tensor)); in run()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dsparsify_gather.cc172 TF_RETURN_IF_ERROR(ckpt_reader->LookupSlice( in ReadTensorFromCheckpoint()