Searched defs:TpuPartitionedCall_Params (Results 1 – 1 of 1) sorted by relevance
107 struct TpuPartitionedCall_Params { struct108 bool input_shape_opt;109 bool group_tensors_for_packing;110 int32_t minimum_input_tensors_packing;111 int32_t minimum_output_tensors_packing;115 bool enable_auto_xla_input_sharding;120 int32_t auto_xla_input_sharding_dim;123 bool enable_variable_deduplication;