Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for shape_value (0.44 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/unfuse_batch_norm_pass.cc

        // Broadcast `multiplier` and `rhs`
        Value shape_value;
        if (!input_type.hasStaticShape()) {
          shape_value = getShapeValue(bn_op.getLoc(), bn_op.getOperand(), rewriter);
        }
        int64_t feature_dim = bn_op.getFeatureIndex();
        auto broadcast_multiplier =
            broadcastToFeatureDim(bn_op.getLoc(), input_type, multiplier,
                                  shape_value, feature_dim, rewriter);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc

    namespace {
    
    Value GetDimValue(OpBuilder &builder, Location loc, Value shape_value,
                      int32_t dim) {
      Type attribute_type = builder.getI64Type();
      return builder.create<TF::StridedSliceOp>(
          loc,
          RankedTensorType::get(
              {}, mlir::cast<ShapedType>(shape_value.getType()).getElementType()),
          /*input=*/shape_value,
          /*begin=*/Create1DConstValue<int32_t>(builder, loc, {dim}),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 13.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

      return success();
    }
    
    // Parses an R1 value to `shape` if it is a TF::ConstOp output. Otherwise,
    // returns an error.
    LogicalResult GetConstShapeValue(Value shape_value,
                                     llvm::SmallVector<int64_t, 8>* shape) {
      auto shape_op = shape_value.getDefiningOp();
      if (!shape_op) return failure();
      auto shape_const_op = llvm::dyn_cast<TF::ConstOp>(shape_op);
      if (!shape_const_op) return failure();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

                   << "output must have the same rank with the length of shape = "
                   << output_rank;
          }
        }
      }
    
      DenseIntElementsAttr shape_value;
      if (matchPattern(shape, m_Constant(&shape_value))) {
        for (const auto shape_elem : shape_value) {
          if (shape_elem.getSExtValue() <= 0) {
            return op.emitOpError("all elements of shape must be > 0");
          }
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

        Value tensor_value = (*input.begin());
        Value shape_value = (*shape.begin());
        new_reorder_op = rewriter.create<ReorderOp>(ods_loc,
                                                    /*tensor=*/tensor_value,
                                                    /*shape=*/shape_value);
        return success();
      }
    
      LogicalResult createFakeQuantOp(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc

      auto reshape_sizes = DenseIntElementsAttr::get(reshape_type, new_shape);
      auto reshape_value =
          builder->create<TF::ConstOp>(input.getLoc(), reshape_sizes);
      return builder->create<TF::ReshapeOp>(input.getLoc(), reshape_result_type,
                                            input, reshape_value);
    }
    
    // Creates transpose op for shape to depth transform.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 29.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        // %shape = [MS1, MS2]
        SmallVector<Value, 4> shape_values;
        shape_values.reserve(input_rank);
        for (int64_t i = 0; i < input_rank; ++i) {
          Value dim_size_value = rewriter.create<mlir::arith::MulIOp>(
              loc, out_dim_size[2 * i], out_dim_size[2 * i + 1]);
          shape_values.push_back(dim_size_value);
        }
        Value shape = rewriter.create<tensor::FromElementsOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  8. pkg/workloadapi/security/authorization.pb.go

    	Scope_WORKLOAD_SELECTOR Scope = 2
    )
    
    // Enum value maps for Scope.
    var (
    	Scope_name = map[int32]string{
    		0: "GLOBAL",
    		1: "NAMESPACE",
    		2: "WORKLOAD_SELECTOR",
    	}
    	Scope_value = map[string]int32{
    		"GLOBAL":            0,
    		"NAMESPACE":         1,
    		"WORKLOAD_SELECTOR": 2,
    	}
    )
    
    func (x Scope) Enum() *Scope {
    	p := new(Scope)
    	*p = x
    	return p
    }
    
    Registered: Fri Jun 14 15:00:06 UTC 2024
    - Last Modified: Fri May 31 02:27:10 UTC 2024
    - 29.7K bytes
    - Viewed (0)
Back to top