Searched refs:value_dtype (Results 1 – 2 of 2) sorted by relevance
433 def check_tensors_dtype_same(data_dtype, value_dtype, op_name): argument435 if value_dtype == data_dtype:
4055 def infer_dtype(self, x_dtype, indices_dtype, value_dtype): argument4057 args = {"x": x_dtype, "value": value_dtype}