Home
last modified time | relevance | path

Searched refs:dynamic_dim_index (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1480 llvm::Optional<int> dynamic_dim_index; in Verify() local
1491 if (dynamic_dim_index) in Verify()
1494 dynamic_dim_index = dim.index(); in Verify()
1500 if (!dynamic_dim_index && total_dim_size != input_dim_size) in Verify()
1506 if (dynamic_dim_index && total_dim_size > input_dim_size) in Verify()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3021 llvm::Optional<int> dynamic_dim_index; in matchAndRewrite() local
3029 assert(!dynamic_dim_index && "invalid split sizes"); in matchAndRewrite()
3030 dynamic_dim_index = dim.index(); in matchAndRewrite()
3044 assert(((dynamic_dim_index && total_dim_size <= input_dim_size) || in matchAndRewrite()
3045 (!dynamic_dim_index && total_dim_size == input_dim_size)) && in matchAndRewrite()
3049 if (dynamic_dim_index) in matchAndRewrite()
3050 split_sizes[*dynamic_dim_index] = input_dim_size - total_dim_size; in matchAndRewrite()