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

  /external/tensorflow/tensorflow/core/kernels/
lookup_util.h 60 TensorShape key_shape = keys_->shape(); local
61 if (!key_shape.IsSameSize(values_->shape())) {
65 key_shape.DebugString(), " vs ", values_->shape().DebugString());
67 if (key_shape.num_elements() == 0) {
initializable_lookup_table.h 67 TensorShape key_shape() const final { return TensorShape(); }
lookup_table_op.cc 123 TensorShape key_shape() const final { return TensorShape(); }
252 TensorShape key_shape() const final { return TensorShape(); }
471 TensorShape key_shape = MaybeVectorizeShape(key_shape_); variable
478 expected_value_shape.RemoveLastDims(key_shape.dims());
492 TensorShape key_shape() const override { return key_shape_; }
672 output_shape.RemoveLastDims(table->key_shape().dims());
    [all...]
  /external/tensorflow/tensorflow/core/framework/
lookup_interface.cc 25 if (!TensorShapeUtils::EndsWith(shape, key_shape())) {
28 key_shape().DebugString());
52 for (int i = 0; i < key_shape().dims(); ++i) {
lookup_interface.h 90 virtual TensorShape key_shape() const = 0;
  /external/tensorflow/tensorflow/contrib/lite/kernels/
hashtable_lookup_test.cc 36 std::initializer_list<int> key_shape,
45 BuildInterpreter({lookup_shape, key_shape, value_shape});
  /external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/
predict_test.cc 38 std::initializer_list<int> key_shape,
52 BuildInterpreter({{input_signature_shape, key_shape, labelweight_shape,
  /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/
sharded_mutable_dense_hashtable.py 87 key_shape = keys.get_shape()
88 if key_shape.ndims > 1:
91 keys = array_ops.slice(keys, [0, 0], [key_shape[0].value, 1])
  /frameworks/ml/nn/common/operations/
HashtableLookupTest.cpp 59 std::initializer_list<uint32_t> key_shape,
71 OperandType KeyTy(Type::TENSOR_INT32, key_shape);

Completed in 195 milliseconds