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