Home
last modified time | relevance | path

Searched refs:set_dimension_size (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis.h228 bool UpdateSetDimensionSizeValueSet(HloInstruction* set_dimension_size);
Dhlo_dataflow_analysis.cc400 HloInstruction* set_dimension_size) { in UpdateSetDimensionSizeValueSet() argument
401 CHECK_EQ(set_dimension_size->opcode(), HloOpcode::kSetDimensionSize); in UpdateSetDimensionSizeValueSet()
403 GetInstructionValueSet(set_dimension_size->operand(0)); in UpdateSetDimensionSizeValueSet()
405 GetInstructionValueSet(set_dimension_size); in UpdateSetDimensionSizeValueSet()
Dhlo_evaluator.h205 Status HandleSetDimensionSize(HloInstruction* set_dimension_size) override;
Dhlo_evaluator.cc438 HloInstruction* set_dimension_size) { in HandleSetDimensionSize() argument
440 GetEvaluatedLiteralFor(set_dimension_size->operand(0)); in HandleSetDimensionSize()
441 Literal result(set_dimension_size->shape()); in HandleSetDimensionSize()
445 GetEvaluatedLiteralFor(set_dimension_size->operand(1)); in HandleSetDimensionSize()
446 result.SetDynamicSize(set_dimension_size->dimension(), in HandleSetDimensionSize()
448 evaluated_[set_dimension_size] = std::move(result); in HandleSetDimensionSize()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dops.mlir1291 func @set_dimension_size(%I: tensor<1x128x512xf32>) -> tensor<1x128x512xf32> {
1295 …%result = "mhlo.set_dimension_size"(%I, %dim) {dimension = 2 : i64} : (tensor<1x128x512xf32>, tens…
1301 func @set_dimension_size(%I: tensor<1x128x512xf32>) -> tensor<1x128x512xf32> {
1305 …%result = "mhlo.set_dimension_size"(%I, %dim) {dimension = 3 : i64} : (tensor<1x128x512xf32>, tens…
Dcanonicalize.mlir978 …%result = "mhlo.set_dimension_size"(%I, %dim) {dimension = 2 : i64} : (tensor<1x128x512xf32>, tens…
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-with-tf2xla.mlir301 // CHECK: mhlo.set_dimension_size
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir501 …%0 = "mhlo.set_dimension_size"(%arg, %size) {dimension = 1 : i64} : (tensor<4x2xf32>, tensor<i32>)…
879 …%0 = "mhlo.set_dimension_size"(%arg, %size) {dimension = 1 : i64} : (tensor<4x4xf32>, tensor<i32>)…
Dimport.hlotxt841 …// CHECK-NEXT: "mhlo.set_dimension_size"([[ARG]], [[SIZE]]) {dimension = 1 : i64} : (tensor<4x4xf…
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops.td1140 def HLO_SetDimensionSizeOp: HLO_Op<"set_dimension_size", [NoSideEffect]>,
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md1687 let padded_v_five: f32[10] = set_dimension_size(v, five, /*dimension=*/0);
1688 let padded_v_six: f32[10] = set_dimension_size(v, six, /*dimension=*/0);