Searched refs:result_rank (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_stitch_op.cc | 116 int64 result_rank = 1 + data0_shape.dims() - indices0_shape.dims(); in Compile() local
118 std::vector<int64> result_shape(result_rank); in Compile()
172 std::vector<int64> slice_start(result_rank); in Compile()
173 std::vector<int64> slice_limit(result_rank); in Compile()
174 std::vector<int64> stride(result_rank, 1); in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | bufferize.cc | 43 int64_t result_rank = result_type.getRank(); in matchAndRewrite() local
44 if (!result_type || !result_type.hasStaticShape() || result_rank > 1) in matchAndRewrite()
51 if (result_rank == 0) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/ |
| D | broadcast_utils.cc | 66 int64_t result_rank = std::max(lhs_type.getRank(), rhs_type.getRank()); in ComputeBinaryElementwiseBroadcastingResultExtents() local
71 loc, RankedTensorType::get({result_rank}, builder.getIndexType()), in ComputeBinaryElementwiseBroadcastingResultExtents()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | hlo_legalize_to_lhlo.cc | 284 auto result_rank = result_type.getRank(); in InsertDynamicMemrefCastOp() local
309 sizes.reserve(result_rank); in InsertDynamicMemrefCastOp()
310 strides.reserve(result_rank); in InsertDynamicMemrefCastOp()
316 for (int i = 0; i < result_rank; ++i) { in InsertDynamicMemrefCastOp()
349 SmallVector<int64_t, 2> dynamic_layout(result_rank, in InsertDynamicMemrefCastOp()
351 SmallVector<int64_t, 2> dynamic_shape(result_rank, in InsertDynamicMemrefCastOp()
|
| D | chlo_legalize_to_hlo.cc | 1095 int64_t result_rank = std::max(lhs_type.getRank(), rhs_type.getRank()); in matchAndRewrite() local
1106 llvm::seq<int64_t>(result_rank - lhs_type.getRank(), result_rank)); in matchAndRewrite()
1114 llvm::seq<int64_t>(result_rank - rhs_type.getRank(), result_rank)); in matchAndRewrite()
|
| D | legalize_to_linalg.cc | 502 int64_t result_rank = shape_type.getDimSize(0); in matchAndRewrite() local
506 for (int i = 0; i < result_rank; ++i) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 438 int64_t result_rank = -1; // -1 means unknown result rank. in matchAndRewrite() local
444 result_rank = element_type.getRank() + 1; in matchAndRewrite()
460 Type shape_type = RankedTensorType::get({result_rank}, shape_dtype); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 3391 int64 result_rank = gather_dim_numbers.offset_dims_size() + in InferGatherShape() local
3396 output_dim_bounds.reserve(result_rank); in InferGatherShape()
3399 output_dim_is_dynamic.reserve(result_rank); in InferGatherShape()
3400 for (int64 i = 0; i < result_rank; i++) { in InferGatherShape()
|