- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 263 for created (0.12 sec)
-
tensorflow/compiler/jit/pjrt_base_device.cc
options.shape_determination_fns) { if (options.shape_determination_fns.empty()) { LOG(ERROR) << "shape_representation_fns must be non-empty."; } VLOG(1) << "Created PJRT base device " << options.compilation_device_name << " device_name: " << name(); } /*static*/ absl::StatusOr<const PjRtBaseDevice::Metadata*> PjRtBaseDevice::GetMetadataFromDevice(DeviceBase* device) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc
!IsQintValueDefinedByIntToQintCast(op->getOperand(i))) { new_operands.push_back(rewriter.create<TF::CastOp>( op->getLoc(), orig_op_type, operands[i])); } else { new_operands.push_back(operands[i]); } } // Create a new UQ op. OperationState state(op->getLoc(), op->getName().getStringRef(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.cc
// Caches a XlaDeviceAllocator per <backend, device ordinal> pair. A // XlaDeviceAllocator is created on demand and is associated with a // XlaDevice. It outlives the device itself (for instance, the buffer // backing a tensor holds a pointer to the allocator for book-keeping, // and this buffer can outlast the device). class XlaDeviceAllocatorState { public: // Creates or returns a cached XlaDeviceAllocator for a given // backend and device_ordinal.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
llvm::SmallVector<mlir::TF::SplitOp, 4> split_ops_for_tiled_input; split_ops_for_tiled_input.reserve( input_sharding.tile_assignment_devices_size()); // Creates a tree of split nodes for sharding tiled inputs. Splits nodes // are created such that input data is sharded in row major order. // Split nodes at ith depth from the original input node represent nodes // that split the input data at i-th dimension.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc
// Set insertion point to right after all operations in cluster. builder->setInsertionPoint(c.ops.back()->getNextNode()); // Create a stand-alone region to hold all instructions in the cluster. Region region; region.push_back(new Block); // Move all operations in cluster to newly created region, stripping their // "device" attribute since launch op already carries device information. Block* block = ®ion.front();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 05 13:30:21 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_annotate_dynamic_shape_inputs.cc
class TPUAnnotateDynamicShapeInputsPass : public impl::TPUAnnotateDynamicShapeInputsPassBase< TPUAnnotateDynamicShapeInputsPass> { void runOnOperation() override; }; // Finds op that created a given value. If the value is a BlockArgument, this // returns the owner of the Block. Operation* GetOpOfValue(Value value) { if (auto block_arg = mlir::dyn_cast<BlockArgument>(value))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_quantization_unit_loc.cc
return isa<NameLoc>(loc) || (isa<CallSiteLoc>(loc) && isa<NameLoc>(mlir::cast<CallSiteLoc>(loc).getCallee())); }); } // Finds the pattern of the location created by `ImporterBase::GetLocation` // in `tensorflow/compiler/mlir/tensorflow/translate/import_model.cc`. void FindQuantizationUnitsRecursively(Location loc, SmallVector<QuantizationUnit>& units) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_to_island.cc
builder.setInsertionPoint(&terminator); builder.create<tf_executor::YieldOp>(terminator.getLoc(), terminator.getOperands()); terminator.erase(); builder.setInsertionPoint(island_op); IRMapping mapping; for (int i : llvm::seq<int>(0, num_replicas)) { // Create new island for replica. auto replica = builder.create<tf_executor::IslandOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 24 21:01:40 UTC 2023 - 16.9K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
TF_Run(session, run_options, nullptr, nullptr, 0, nullptr, nullptr, 0, nullptr, 0, run_metadata, s); EXPECT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(s)) << TF_Message(s); EXPECT_EQ("Session was not created with a graph before Run()!", string(TF_Message(s))); TF_DeleteBuffer(run_metadata); TF_DeleteBuffer(run_options); TF_DeleteDeprecatedSession(session, s);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 96.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc
// Create tensor attribute for this variable. absl::StatusOr<ElementsAttr> tensor_attr_or = ConvertTensor(tensor, &builder); if (!tensor_attr_or.ok()) { return module.emitOpError() << "failed to convert tensor (name: " << name.str() << ")"; } ElementsAttr tensor_attr = tensor_attr_or.value(); builder.create<tf_saved_model::GlobalTensorOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0)