Searched defs:key_shape (Results 1 – 8 of 8) sorted by relevance
71 TensorShape key_shape() const final { return TensorShape(); } in key_shape() function179 TensorShape key_shape = keys_->shape(); in KeyValueTensorIterator() local
25 TensorShape const& key_shape, in CheckKeyCounterShape()
143 TensorShape key_shape() const final { return TensorShape(); } in key_shape() function in tensorflow::lookup::MutableHashTableOfScalars292 TensorShape key_shape() const final { return TensorShape(); } in key_shape() function in tensorflow::lookup::MutableHashTableOfTensors557 TensorShape key_shape = MaybeVectorizeShape(key_shape_); in CheckKeyAndValueTensorsForImport() local578 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()
132 const xla::Shape key_shape = in Compile() local
2171 Shape key_shape = sort->operand(0)->shape(); in HandleSort() local
4812 std::initializer_list<int> key_shape, in HashtableLookupOpModel()