Searched refs:target_dim_num (Results 1 – 4 of 4) sorted by relevance
74 int64 target_dim_num = binding.target_param_dim_num(); in CreateFromProto() local79 target_dim_num})); in CreateFromProto()
320 // A list of bindings which indicates that the `target_dim_num` in
301 int64 target_dim_num) { in SetDynamicBinding() argument313 param_shape_ptr->set_dynamic_dimension(target_dim_num, in SetDynamicBinding()331 target_param_num, target_param_index, target_dim_num})); in SetDynamicBinding()
330 ShapeIndex target_param_index, int64 target_dim_num);