Searched defs:key_shape (Results 1 – 7 of 7) sorted by relevance
72 TensorShape key_shape() const final { return TensorShape(); } in key_shape() function225 TensorShape key_shape = keys_->shape(); in KeyValueTensorIterator() local
25 TensorShape const& key_shape, in CheckKeyCounterShape()
142 TensorShape key_shape() const final { return TensorShape(); } in key_shape() function in tensorflow::lookup::MutableHashTableOfScalars330 TensorShape key_shape() const final { return TensorShape(); } in key_shape() function in tensorflow::lookup::MutableHashTableOfTensors644 TensorShape key_shape = MaybeVectorizeShape(key_shape_); in CheckKeyAndValueTensorsForImport() local665 TensorShape key_shape() const override { return key_shape_; } in key_shape() function in tensorflow::lookup::MutableDenseHashTable
205 TensorShape key_shape = it->first.shape(); in Compute() local
42 std::initializer_list<int> key_shape, in HashtableLookupOpModel()
2235 Shape key_shape = sort->operand(0)->shape(); in HandleSort() local
4812 std::initializer_list<int> key_shape, in HashtableLookupOpModel()