Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 46 of 46 for shape_of (0.17 sec)

  1. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

      std::string serialized_func_module;
      if (HasDynamicOutputs(external_outputs)) {
        FuncOp shape_op = BuildFunction(clustered_ops, external_operands,
                                        external_outputs, &builder);
        EncapsulateFuncAndSerialize(shape_op, &serialized_func_module);
      }
    
      builder.setInsertionPoint(&op);
      SmallVector<Value, 4> host_compute_outputs;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir

        %tl_1 = "tf.TensorListPushBack"(%tl_0, %elem) : (tensor<!tf_type.variant<tensor<?x1xf32>>>, tensor<16x1xf32>) -> tensor<!tf_type.variant<tensor<?x1xf32>>>
        %shape_32 = "tf.TensorListElementShape"(%tl_1) : (tensor<!tf_type.variant<tensor<?x1xf32>>>) -> tensor<?xi32>
        %shape_64 = "tf.TensorListElementShape"(%tl_1) : (tensor<!tf_type.variant<tensor<?x1xf32>>>) -> tensor<?xi64>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 23 17:24:10 UTC 2024
    - 167.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

          return ComputeOutputComponent(ValuePort(id.getInput()), values);
        return nullptr;
      }
    
      if (auto shape_op = dyn_cast<TF::ShapeOp>(op)) {
        // No shape available in an unranked tensor type.
        auto operand_ty =
            mlir::dyn_cast<RankedTensorType>(shape_op.getOperand().getType());
        if (!operand_ty) return nullptr;
    
        // Shape op has a single output so the first element should always be zero
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

          shape.push_back(dim.getSExtValue());
        }
        return tensorflow::GetTypeFromTFTensorShape(shape, etype);
      }
    
      if (auto shape_op = dims.getDefiningOp<ShapeOp>()) {
        if (auto t = mlir::dyn_cast<ShapedType>(shape_op.getInput().getType())) {
          return t;
        }
      }
    
      return UnrankedTensorType::get(etype);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

    absl::StatusOr<mlir::Type> ImporterBase::InferOutputType(
        const Node& node, int idx, mlir::Builder builder) {
      DataType dtype = node.properties()->output_types[idx];
    
      // Returns output type given inference context.
      auto shape_ic =
          [&](shape_inference::InferenceContext* c) -> absl::StatusOr<mlir::Type> {
        // TODO(b/200093974): Post triage, consider following
        // GraphConstructor::ValidateShape in checking _output_shapes always.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/schema/schema_generated.h

        const std::vector<int32_t> *shape = nullptr,
        tflite::TensorType type = tflite::TensorType_FLOAT32,
        bool has_rank = false) {
      auto shape__ = shape ? _fbb.CreateVector<int32_t>(*shape) : 0;
      return tflite::CreateVariantSubType(
          _fbb,
          shape__,
          type,
          has_rank);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 1M bytes
    - Viewed (0)
Back to top