Search Options

Results per page
Sort
Preferred Languages
Advance

Results 111 - 120 of 242 for get_attr (0.15 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

        }
        OpBuilder builder(call);
        auto new_call = builder.create<CallOp>(
            call.getLoc(), info.decomposed_callee.getFunctionType().getResults(),
            new_operands, call->getAttrs());
        new_call->setAttr(
            "f", SymbolRefAttr::get(
                     builder.getContext(),
                     const_cast<func::FuncOp&>(info.decomposed_callee).getName()));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc

      EXPECT_TRUE(isa<TF::XlaCallModuleOp>(lifted_op));
      EXPECT_EQ(
          mlir::cast<StringAttr>(lifted_op->getAttr("_original_entry_function")),
          "composite_dot_general_fn_1");
      EXPECT_EQ(
          mlir::cast<ArrayAttr>(lifted_dot_general_op->getAttr("precision_config")),
          builder_.getArrayAttr(SmallVector<Attribute>(
              1, mlir::stablehlo::PrecisionAttr::get(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 26.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/utils/device_util.cc

          devices.push_back(builder.getNamedAttr(name, metadata));
        }
      }
    
      op->setAttr(kDevicesAttr, builder.getDictionaryAttr(devices));
    }
    
    mlir::LogicalResult GetDevicesFromOp(mlir::Operation* op,
                                         mlir::TF::RuntimeDevices* devices) {
      auto devices_attr = op->getAttr(kDevicesAttr);
      if (!devices_attr) return mlir::success();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc

        if (auto device = res->getAttrOfType<StringAttr>("device")) {
          if (!device.getValue().empty())
            result_id->setAttr("device", device);
          else
            result_id->setAttr("device", compile_device_op);
        } else if (compile_device_op) {
          result_id->setAttr("device", compile_device_op);
        }
        res.getOutput().replaceAllUsesWith(compile_op->getResult(0));
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 29.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/custom_op_with_tflite_op.mlir

      // tf.MyCustomOp is the result of conversion to a Custom op
      %2 = "tf.MyCustomOp"(%1, %0) {fused_activation_function = "RELU", int_attr = 2 : i32}  : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32> loc("MyCustomOp")
      %3 = "tfl.exp"(%2)  : (tensor<4xf32>) -> tensor<4xf32> loc("exp")
      func.return %3 : tensor<4xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 14 16:41:28 UTC 2022
    - 4.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/utils/fake_quant_utils.cc

        // TODO(fengliuai): copy the regions.
        OperationState state(op->getLoc(), op->getName().getStringRef(),
                             parent_op->getOperands(), parent_op->getResultTypes(),
                             op->getAttrs(), op->getSuccessors());
        Operation* inlined = builder.create(state);
    
        parent_op->replaceAllUsesWith(inlined);
        parent_op->erase();
      }
      return success();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 03 00:14:05 UTC 2023
    - 4.3K bytes
    - Viewed (0)
  7. pkg/registry/resource/resourceclaim/storage/storage.go

    		TableConvertor: printerstorage.TableConvertor{TableGenerator: printers.NewTableGenerator().With(printersinternal.AddHandlers)},
    	}
    	options := &generic.StoreOptions{RESTOptions: optsGetter, AttrFunc: resourceclaim.GetAttrs}
    	if err := store.CompleteWithOptions(options); err != nil {
    		return nil, nil, err
    	}
    
    	statusStore := *store
    	statusStore.UpdateStrategy = resourceclaim.StatusStrategy
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri Nov 18 10:11:16 UTC 2022
    - 3.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc

            rewriter.getDenseI64ArrayAttr(kOihwToHwioPermutation));
    
        // [b, 0, 1, f]x[0, 1, i, o]->[b, 0, 1, f]
        const auto new_dimension_nums = rewriter.getAttr<ConvDimensionNumbersAttr>(
            /*inputBatchDimension=*/0, /*inputFeatureDimension=*/3,
            /*inputSpatialDimensions=*/SmallVector<int64_t>{1, 2},
            /*kernelInputFeatureDimension=*/2, /*kernelOutputFeatureDimension=*/3,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  9. tensorflow/c/ops.cc

        cc_type v;                                                           \
        auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx);             \
        Status s = cc_ctx->GetAttr(attr_name, &v);                           \
        Set_TF_Status_from_Status(status, s);                                \
        if (s.ok()) {                                                        \
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 28 22:41:35 UTC 2022
    - 10.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc

      for (const auto& attr_def : op_reg_data->op_def.attr()) {
        registered_attrs.insert(attr_def.name());
      }
      // Attributes are not in the registered attributes set will be ignored.
      for (auto& attr : inst->getAttrs()) {
        if (registered_attrs.find(attr.getName()) == registered_attrs.end()) {
          attrs_to_ignore->insert(
              absl::string_view(attr.getName().data(), attr.getName().size()));
        }
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 11.1K bytes
    - Viewed (0)
Back to top