HomeSort by relevance Sort by last modified time
    Searched refs:LookupDtypeAndShape (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/c/
checkpoint_reader.cc 98 status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape);
  /external/tensorflow/tensorflow/core/util/tensor_bundle/
tensor_bundle.h 198 Status LookupDtypeAndShape(StringPiece key, DataType* dtype,
212 // extra allocation. These can be queried via "LookupDtypeAndShape()".
tensor_bundle_test.cc 58 // Tests for LookupDtypeAndShape().
61 TF_ASSERT_OK(reader->LookupDtypeAndShape(key, &dtype, &shape));
75 // Tests for LookupDtypeAndShape().
78 TF_ASSERT_OK(reader->LookupDtypeAndShape(key, &dtype, &shape));
tensor_bundle.cc     [all...]
  /external/tensorflow/tensorflow/core/kernels/
load_and_remap_matrix_op.cc 139 OP_REQUIRES_OK(context, reader.LookupDtypeAndShape(

Completed in 483 milliseconds