- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for total_dim_size (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
} else { total_dim_size += dim_val; } } if (!dynamic_dim_index && total_dim_size != input_dim_size) return op.emitOpError( "split sizes must sum up to the dimension size along split " "dimension, found ") << total_dim_size << " vs " << input_dim_size; if (dynamic_dim_index && total_dim_size > input_dim_size)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
assert(((dynamic_dim_index && total_dim_size <= input_dim_size) || (!dynamic_dim_index && total_dim_size == input_dim_size)) && "invalid split sizes"); // Update the dynamic dimension with calculated concrete size. if (dynamic_dim_index) split_sizes[*dynamic_dim_index] = input_dim_size - total_dim_size; // Parameters for constructing each slice.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)