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.cc622 TF_ASSERT_OK(reader.LookupSlice("foo", distinct_slice, &val)); in TEST()
632 TF_ASSERT_OK(reader.LookupSlice("foo", distinct_slice, &val)); in TEST()
656 TF_ASSERT_OK(reader.LookupSlice("no_extents", slice, &val)); in TEST()
665 TF_ASSERT_OK(reader.LookupSlice("both_extents", slice, &val)); in TEST()
674 TF_ASSERT_OK(reader.LookupSlice("no_extents", slice, &val)); in TEST()
683 TF_ASSERT_OK(reader.LookupSlice("both_extents", slice, &val)); in TEST()
Dtensor_bundle.h244 Status LookupSlice(StringPiece full_tensor_key, const TensorSlice& slice_spec,
Dtensor_bundle.cc983 Status BundleReader::LookupSlice(StringPiece full_tensor_key, in LookupSlice() function in tensorflow::BundleReader
/external/tensorflow/tensorflow/core/kernels/
Dload_and_remap_matrix_op.cc212 OP_REQUIRES_OK(context, reader.LookupSlice(old_tensor_name, tensor_slice, in Compute()
Dsave_restore_tensor.cc302 reader->LookupSlice(tensor_name, parsed_slice, restored_tensor)); in run()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dsparsify_gather.cc171 TF_RETURN_IF_ERROR(ckpt_reader->LookupSlice( in ReadTensorFromCheckpoint()