Searched refs:set_dynamic_dimension_size (Results 1 – 3 of 3) sorted by relevance
298 // CHECK-LABEL: @set_dynamic_dimension_size299 func @set_dynamic_dimension_size(%input: tensor<4xf32>, %size: tensor<i32>) -> tensor<4xf32> {
416 set_dynamic_dimension_size = gen_xla_ops.xla_set_dynamic_dimension_size variable
1410 padding_map.arg_index] = tf2xla.set_dynamic_dimension_size(