Searched refs:output_sparse_shapes_tensor (Results 1 – 2 of 2) sorted by relevance
325 std::vector<Tensor>* output_sparse_shapes_tensor) { in BatchExampleProtoToTensors() argument385 (*output_sparse_shapes_tensor)[d] = in BatchExampleProtoToTensors()388 auto shape_t = (*output_sparse_shapes_tensor)[d].vec<int64>(); in BatchExampleProtoToTensors()
117 std::vector<Tensor>* output_sparse_shapes_tensor);