Search Options

Results per page
Sort
Preferred Languages
Advance

Results 51 - 60 of 61 for if_builder (0.22 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/optimize.td

    def IsDataFormatNHWC : ConstantAttr<TF_ConvnetDataFormatAttr, "\"NHWC\"">;
    
    // Get the last dimension size as a 1-d single element attr.
    def GetLastDimSizeAsI32 : NativeCodeCall<
      "DenseElementsAttr::get(RankedTensorType::get({1}, $_builder.getIntegerType(32)), "
      "static_cast<int32_t>($0.getType().cast<RankedTensorType>().getDimSize(  "
      "  $0.getType().cast<RankedTensorType>().getRank() - 1)))">;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 22 07:31:23 UTC 2023
    - 5.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc

                                  shape_determination_fns, compilation_result);
    }
    
    static mlir::LogicalResult MlirTfToHloTextTranslateFunctionImpl(
        mlir::ModuleOp module_op, llvm::raw_ostream& output, bool via_builder) {
      if (!module_op) return mlir::failure();
    
      llvm::SmallVector<TensorOrResourceShape, 4> arg_shapes;
      auto args_status =
          ParseArgumentShapes(mlir::StringRefToView(input_shapes), arg_shapes);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 18.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td

    `tf_device` dialect operation.
      }];
    
      let arguments = (ins
        Variadic<AnyType>:$results
      );
    
      let builders = [
        OpBuilder<(ins),
        [{
          build($_builder, $_state, {});
        }]>
       ];
    
      let assemblyFormat = "attr-dict ($results^ `:` type($results))?";
    }
    
    def TfDevice_LaunchFuncOp : TfDevice_Op<"launch_func", []> {
      let summary = [{
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 23 23:53:20 UTC 2024
    - 14.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

            tensors. Each tensor contains the asset file names.
        """
        v1_builder = builder.SavedModelBuilder(saved_model_path)
        sig_def = signature_def_utils_impl.predict_signature_def(
            inputs=inputs, outputs=outputs
        )
    
        v1_builder.add_meta_graph_and_variables(
            sess,
            tags,
            signature_def_map={signature_key: sig_def},
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td

       (HasNonNegativeOffset $begin, $end)]>;
    
    def GetNumElementsOrOne: NativeCodeCall<"GetNumElementsOrOne($0.getType())">;
    
    def ReshapeValueDroppingLastDim : NativeCodeCall<
      "ReshapeValueDroppingLastDim($_builder, $0)">;
    
    def IsOneHotIndexAttribute : Constraint<CPred<
      "TFL::IsOneHotIndexAttribute($0)">>;
    
    // Checks if the shape has static shape with last dimension equals 1.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 20:31:41 UTC 2024
    - 66.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        "StringAttr":$fusedActivationFunction),
      [{
        buildFusedBroadcastableBinOp(
           &$_builder, $_state, lhs, rhs, fusedActivationFunction);
      }]>;
    
    def TFL_ComparisonBinaryBuilder :
      OpBuilder<(ins "Value":$lhs, "Value":$rhs),
      [{
        buildComparisonBinOp(&$_builder, $_state, lhs, rhs);
      }]>;
    
    //===----------------------------------------------------------------------===//
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tfr/ir/tfr_ops.td

    // tensor argument types
    class TFR_Type<string name> : DialectType<TFR_Dialect,
        CPred<"$_self.isa<mlir::TFR::" # name # "Type>()">,
        "TFR " # name #" type">,
        BuildableType<"$_builder.getType<mlir::TFR::" # name # "Type>()">;
    def TFR_TensorType : TFR_Type<"TFRTensor">;
    def TFR_TensorListType : TFR_Type<"TFRTensorList">;
    def TFR_AllTensorTypes : Type<Or<[
        TFR_TensorType.predicate,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 10:54:29 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

          )
          signature_map = {'main': signature}
    
          tags = {tag_constants.SERVING}
          v1_builder = builder.SavedModelBuilder(self._input_saved_model_path)
          v1_builder.add_meta_graph_and_variables(
              sess, tags, signature_def_map=signature_map
          )
          v1_builder.save()
    
        quantization_options = quant_opts_pb2.QuantizationOptions(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
  9. src/cmd/go/internal/work/exec.go

    	}
    
    	if argLen > sys.ExecArgLengthLimit {
    		return true
    	}
    
    	// On the Go build system, use response files about 10% of the
    	// time, just to exercise this codepath.
    	isBuilder := os.Getenv("GO_BUILDER_NAME") != ""
    	if isBuilder && rand.Intn(10) == 0 {
    		return true
    	}
    
    	return false
    }
    
    // encodeArg encodes an argument for response file writing.
    func encodeArg(arg string) string {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Mon Jun 10 14:46:37 UTC 2024
    - 105.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

          "llvm::ArrayRef<::mlir::NamedAttribute>":$attributes,
          "unsigned":$numRegions),
        [{
          assert(numRegions == 2u && "mismatched number of regions");
          build($_builder, $_state, resultTypes, operands, attributes);
        }]>];
    
      let hasCanonicalizer = 1;
    }
    
    def TF_GeneratorDatasetRegionOp : TF_Op<"GeneratorDatasetRegion",
          [AttrSizedOperandSegments,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
Back to top