Searched refs:sub_elements (Results 1 – 3 of 3) sorted by relevance
232 const int64_t sub_elements = in UpdateOutputVariables() local234 if (sub_elements != output_tensor_shape_protos.size()) { in UpdateOutputVariables()236 "Mismatched numbers of output shapes: ", sub_elements, " vs. ", in UpdateOutputVariables()240 if (sub_elements != variables.size()) { in UpdateOutputVariables()242 "Output count does not equal varaible count: ", sub_elements, " vs. ", in UpdateOutputVariables()247 output_tensor_shapes.reserve(sub_elements); in UpdateOutputVariables()248 for (int64_t i = 0; i < sub_elements; ++i) { in UpdateOutputVariables()
403 const int64_t sub_elements = in AllocateOutputTensors() local405 if (sub_elements != output_tensor_shape_protos.size()) { in AllocateOutputTensors()407 "Mismatched numbers of output shapes: ", sub_elements, " vs. ", in AllocateOutputTensors()415 output_tensor_shapes.reserve(sub_elements); in AllocateOutputTensors()416 for (int64_t i = 0; i < sub_elements; ++i) { in AllocateOutputTensors()446 for (int64_t i = 0; i < sub_elements; ++i) { in AllocateOutputTensors()
30 int sub_elements = 1; in GetFlattenedIndex() local32 index += indices[i] * sub_elements; in GetFlattenedIndex()33 sub_elements *= shape[i]; in GetFlattenedIndex()