Search Options

Results per page
Sort
Preferred Languages
Advance

Results 91 - 100 of 182 for created (0.17 sec)

  1. 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.
    TF::TransposeOp GetTransposeOpForConv2DFilter(OpBuilder* builder, Value input) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 29.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/lift_quantizable_spots_as_functions.cc

        if (failed(ApplyQuantizationSpec(spec, module_op))) {
          signalPassFailure();
          return;
        }
      }
    }
    
    }  // namespace
    
    // Creates `LiftQuantizableSpotsAsFunctionsPass` with user-defined
    // `QuantizationSpecs`.
    std::unique_ptr<OperationPass<ModuleOp>>
    CreateLiftQuantizableSpotsAsFunctionsPass(
        const QuantizationSpecs& quantization_specs) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 9.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc

          } else {
            if (IsCompatibleTypeWithTFLCastOp(value.getType()) &&
                IsCompatibleTypeWithTFLCastOp(type)) {
              auto cast = b.create<CastOp>(yield_op->getLoc(), type, value);
              args.push_back(cast);
            } else {
              auto cast = b.create<TF::CastOp>(yield_op->getLoc(), type, value);
              args.push_back(cast);
            }
          }
        }
        args.append(new_args.begin(), new_args.end());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc

    }
    
    // Creates a wrapper island enclosing the `sub_op` dependent on
    // `control_inputs`.
    IslandOp CreateIsland(Operation* sub_op, ValueRange control_inputs,
                          OpBuilder builder) {
      assert(sub_op);
      auto control_type = ControlType::get(builder.getContext());
      auto island = builder.create<IslandOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc

            auto s_qconst =
                builder.create<SparseQConstOp>(op->getLoc(), cst.getQtypeAttr(),
                                               cst.getValue(), s_param, new_value);
            value.replaceAllUsesWith(s_qconst.getResult());
            cst.erase();
          }
    
          if (result.needs_densify) {
            auto value = op->getOperand(operand);
            auto densify =
                builder.create<DensifyOp>(op->getLoc(), value.getType(), value);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc

          RankedTensorType::get(static_cast<int32_t>(1), builder.getI32Type());
      auto out_segids_cst = builder.create<TFL::ConstOp>(
          builder.getI32TensorAttr(flattened_out_segids));
      auto contracting_segids_cst = builder.create<TFL::ConstOp>(
          builder.getI32TensorAttr(flattened_contracting_segids));
      auto num_segids_tensor =
          builder.create<TFL::ConstOp>(DenseIntElementsAttr::get(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/flags.cc

               "Disable deadness related safety checks when clustering (this is "
               "unsound)."),
          Flag("tf_xla_disable_resource_variable_safety_checks_for_debugging",
               &mark_for_compilation_flags
                    ->tf_xla_disable_resource_variable_safety_checks_for_debugging,
               "Disable resource variables related safety checks when clustering "
               "(this is unsound)."),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 17 18:52:57 UTC 2024
    - 24.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

      if (!shaped_type) {
        return errors::Internal("Constant doesn't have a shape");
      }
      auto op = builder.create<tfl::ExternalConstOp>(
          loc, shaped_type, builder.getI32IntegerAttr(buffer_index));
      return op.getOperation();
    }
    
    // TODO(b/172664358): Creates a new op instead of reusing constant op.
    // Creates a constant op with "tfl.is_variable" attribute to represent stateful
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc

        return a->isBeforeInBlock(b);
      });
    
      // Create tf_device::ClusterOp before the last operation in the block that
      // is a part of a match set.
      auto back = cluster.operations.back();
      auto loc = back->getLoc();
      OpBuilder builder(back);
    
      auto cluster_op =
          builder.create<tf_device::ClusterOp>(loc, return_types, policy);
    
      // Create block in cluster_op's region and move 'cluster.operations' into
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 27.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

        auto slice_size_attr = DenseIntElementsAttr::get(idx_type, slice_size_i32);
        auto slice_size =
            rewriter.create<arith::ConstantOp>(op.getLoc(), slice_size_attr);
    
        ArrayRef<int64_t> strides = op.getStrides();
        // If stride of every dimension is 1, create tfl.slice and return early.
        // Otherwise, create tfl.strided_slice instead.
        if (llvm::all_of(strides, [](int64_t stride) { return stride == 1; })) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
Back to top