Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for is_constant (0.16 sec)

  1. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

                                             /*is_constant=*/true));
      auto shaped_type = type.dyn_cast<mlir::RankedTensorType>();
      if (!shaped_type) {
        return errors::Internal("Constant doesn't have a shape");
      }
    
      TF_ASSIGN_OR_RETURN(type, tfl::GetTensorType(tensor, builder,
                                                   /*is_constant=*/true,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

                                  options.op_registry(), &debug_info);
        enter_builder.Attr("frame_name",
                           control_flow_info[src_node->id()].frame_name);
        enter_builder.Attr("is_constant", true);
        enter_builder.Input(node, 0);
        Node* enter_node = options.FinalizeBuilder(&enter_builder);
        // Adopt the new Enter node as the value in the current frame.
        node = enter_node;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/deadness_analysis_test.cc

          root.WithOpName("capture_enter_outer"), captured, "outer_loop",
          ops::internal::Enter::Attrs().IsConstant(true));
      Output capture_enter_inner = ops::internal::Enter(
          root.WithOpName("capture_enter_inner"), capture_enter_outer, "inner_loop",
          ops::internal::Enter::Attrs().IsConstant(true));
      Output mul0 = ops::Mul(root.WithOpName("mul0"), div1_inner.induction_var,
                             capture_enter_inner);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 51.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

        if (!begin_attr_type ||
            !matchPattern(begin_attr, m_Constant(&begin_elem_attr))) {
          return failure();
        }
        if (!end_attr_type || !matchPattern(end_attr, m_Constant(&end_elem_attr))) {
          return failure();
        }
        if (!strides_attr_type ||
            !matchPattern(strides_attr, m_Constant(&strides_elem_attr))) {
          return failure();
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

        ElementsAttr paddings;
        llvm::SmallVector<int64_t, 4> block_shape_ints;
        auto padded_shape = llvm::to_vector<4>(input_shape);
        if (matchPattern(op.getBlockShape(), m_Constant(&block_shape)) &&
            matchPattern(op.getPaddings(), m_Constant(&paddings))) {
          for (uint64_t i = 0; i < block_rank; i++) {
            int64_t paddings_sum =
                paddings.getValues<APInt>()[{i, 0}].getSExtValue() +
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

                         : op.emitOpError(error_info);
            }
          }
        }
    
        DenseIntElementsAttr dense_elem_attr;
        if (matchPattern(element_shape, m_Constant(&dense_elem_attr))) {
          // Note: It's technically unsafe to rewrite
          //     TensorListReserve(num_element, element_shape)
          // to
          //     Fill(Concat(num_element, element_shape), 0)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/mark_for_compilation_pass.cc

        return false;
      }
    
      Node* const_input = nullptr;
      for (const Edge* e : n->in_edges()) {
        if (!e->IsControlEdge() && e->src()->IsConstant()) {
          const_input = e->src();
          break;
        }
      }
    
      if (!const_input) {
        return false;
      }
    
      const TensorProto* proto = nullptr;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

     private:
      LogicalResult matchAndRewrite(quantfork::QuantizeCastOp q_op,
                                    PatternRewriter& rewriter) const override {
        DenseFPElementsAttr attr;
        if (!matchPattern(q_op.getArg(), m_Constant(&attr))) {
          return failure();
        }
    
        ShapedType tensor_qtype =
            mlir::cast<ShapedType>(q_op.getResult().getType());
        Attribute tensor_proto_attr = Quantize(attr, tensor_qtype);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
Back to top