Searched refs:coo_sparse (Results 1 – 7 of 7) sorted by relevance
72 tensor_info.coo_sparse.values_tensor_name = tensor.values.name73 tensor_info.coo_sparse.indices_tensor_name = tensor.indices.name74 tensor_info.coo_sparse.dense_shape_tensor_name = tensor.dense_shape.name148 _get_tensor(tensor_info.coo_sparse.indices_tensor_name),149 _get_tensor(tensor_info.coo_sparse.values_tensor_name),150 _get_tensor(tensor_info.coo_sparse.dense_shape_tensor_name))
75 x_tensor_info.coo_sparse.values_tensor_name)77 x_tensor_info.coo_sparse.indices_tensor_name)79 x_tensor_info.coo_sparse.dense_shape_tensor_name)
634 tensor_with_coo.coo_sparse.values_tensor_name = "foo"
237 NodeName(input.coo_sparse().values_tensor_name()))) { in GrapplerItemFromMetaGraphDef()241 NodeName(input.coo_sparse().values_tensor_name()), value_tensor); in GrapplerItemFromMetaGraphDef()245 NodeName(input.coo_sparse().indices_tensor_name()))) { in GrapplerItemFromMetaGraphDef()249 NodeName(input.coo_sparse().indices_tensor_name()), in GrapplerItemFromMetaGraphDef()254 NodeName(input.coo_sparse().dense_shape_tensor_name()))) { in GrapplerItemFromMetaGraphDef()258 NodeName(input.coo_sparse().dense_shape_tensor_name()), in GrapplerItemFromMetaGraphDef()285 NodeName(output.coo_sparse().values_tensor_name()))) { in GrapplerItemFromMetaGraphDef()287 NodeName(output.coo_sparse().values_tensor_name())); in GrapplerItemFromMetaGraphDef()291 NodeName(output.coo_sparse().indices_tensor_name()))) { in GrapplerItemFromMetaGraphDef()293 NodeName(output.coo_sparse().indices_tensor_name())); in GrapplerItemFromMetaGraphDef()[all …]
41 const TensorInfo_CooSparse& coo_sparse = tensor_info.coo_sparse(); in GetTensorNamesFromTensorInfo() local42 tensor_names->insert(coo_sparse.values_tensor_name()); in GetTensorNamesFromTensorInfo()43 tensor_names->insert(coo_sparse.indices_tensor_name()); in GetTensorNamesFromTensorInfo()44 tensor_names->insert(coo_sparse.dense_shape_tensor_name()); in GetTensorNamesFromTensorInfo()
13 name: "coo_sparse"
235 CooSparse coo_sparse = 4; field