Searched refs:sparse_tensor_components (Results 1 – 1 of 1) sorted by relevance
89 const std::vector<SparseTensorToComponentInfo>& sparse_tensor_components) { in UpdateFunctionInputAttributes() argument104 for (const auto component : sparse_tensor_components) { in UpdateFunctionInputAttributes()114 for (const auto component : sparse_tensor_components) { in UpdateFunctionInputAttributes()138 std::vector<SparseTensorToComponentInfo>* sparse_tensor_components) { in CreateComponentTensorsFromSparseTensors() argument143 sparse_tensor_components->push_back(SparseTensorToComponentInfo{ in CreateComponentTensorsFromSparseTensors()190 std::vector<SparseTensorToComponentInfo> sparse_tensor_components; in runOnOperation() local192 &sparse_tensor_components); in runOnOperation()198 sparse_tensor_components) { in runOnOperation()230 for (int i = 0; i < sparse_tensor_components.size(); ++i) in runOnOperation()232 sparse_tensor_components[i].func_op_arg_index - i); in runOnOperation()[all …]