Searched defs:key_shape (Results 1 – 7 of 7) sorted by relevance
69 TensorShape key_shape() const final { return TensorShape(); } in key_shape() function171 TensorShape key_shape = keys_->shape(); in KeyValueTensorIterator() local
132 TensorShape key_shape() const final { return TensorShape(); } in key_shape() function in tensorflow::lookup::MutableHashTableOfScalars270 TensorShape key_shape() const final { return TensorShape(); } in key_shape() function in tensorflow::lookup::MutableHashTableOfTensors535 TensorShape key_shape = MaybeVectorizeShape(key_shape_); in CheckKeyAndValueTensorsForImport() local556 TensorShape key_shape() const override { return key_shape_; } in key_shape() function in tensorflow::lookup::MutableDenseHashTable
36 std::initializer_list<int> key_shape, in HashtableLookupOpModel()
38 std::initializer_list<int> key_shape, in PredictOpModel()
126 const xla::Shape key_shape = in Compile() local
1451 Shape key_shape = sort->operand(0)->shape(); in HandleSort() local
3521 std::initializer_list<int> key_shape, in HashtableLookupOpModel()