Searched defs:input_a_tensor_map (Results 1 – 2 of 2) sorted by relevance
232 void TensorDotInfo::InferTensorMapAxesTuple(size_t size, const TensorMap &input_a_tensor_map, in InferTensorMapAxesTuple()284 TensorMap tensor_map_index, input_a_tensor_map; in InferTensorMap() local
383 Shape input_a_tensor_map = inputs_tensor_map_.at(0); in InferMirrorOps() local