- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 226 for pushBack (0.11 sec)
-
tensorflow/compiler/mlir/lite/flatbuffer_operator.cc
if (value.IsBool()) { mlir_vector.push_back(BuildVhloBooleanV1Attr(value.AsBool(), builder)); } else if (value.IsString()) { mlir_vector.push_back( BuildVhloStringV1Attr(value.AsString().str(), builder)); } else if (value.IsInt()) { mlir_vector.push_back(BuildVhloIntV1Attr(value.AsInt64(), builder)); } else if (value.IsFloat()) { mlir_vector.push_back(BuildVhloFloatV1Attr(value.AsFloat(), builder));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 38K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
input_arguments.push_back(tensor_i32); input_arguments.push_back(one_matrix->getResult(0)); lhs_rank = tensor_shape.size(); rhs_rank = opponent_shape.size(); } else { output_shape = CreateOutputShape(reduce_dnums, opponent_shape, tensor_shape); input_arguments.push_back(one_matrix->getResult(0)); input_arguments.push_back(tensor_i32); lhs_rank = opponent_shape.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/common/case_format.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 09:51:28 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_island_coarsening.cc
control_fetches.push_back(value); } else { data_fetches.push_back(value); data_types.push_back(value.getType()); } } auto island = OpBuilder(fetch).create<IslandOp>( fetch.getLoc(), data_types, /*control=*/ControlType::get(fetch.getContext()), /*controlInputs=*/control_fetches); island.getBody().push_back(new Block);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
transpose_ops.push_back(transpose); } } // Nothing to do here. if (!permutation_op || transpose_ops.empty()) return; SmallVector<int64_t, 4> permutation; auto perm_attr = mlir::cast<DenseElementsAttr>(permutation_op.getValue()); for (const auto& value : perm_attr.getValues<APInt>()) permutation.push_back(value.getSExtValue());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
sharding_for_args.push_back(operand_sharding); continue; } if (infer_from_computation) { auto arg_sharding = GetXlaShardingFromArg(arg, logical_device_vec); if (arg_sharding) { sharding_for_args.push_back(arg_sharding.value()); continue; } } sharding_for_args.push_back(std::nullopt); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
builder.setInsertionPoint(call); auto new_operands = llvm::to_vector(call.getArgOperands()); new_operands.push_back(token); auto new_result_types = llvm::to_vector(call.getResultTypes()); new_result_types.push_back(token.getType()); auto new_call = builder.create<func::CallOp>( call.getLoc(), new_result_types, new_symbol ? *new_symbol : call.getCallee(), new_operands);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter.cc
if (!main_fn) return std::string(""); flatbuffers::FlatBufferBuilder fb_builder; std::vector<mlir::func::FuncOp> funcs; funcs.push_back(main_fn); module.walk([&](mlir::func::FuncOp fn) { if (fn != main_fn) { funcs.push_back(fn); } }); // Populate the hardware metadata. // And collect the hardwares used. std::map<std::string, uint8_t> hardware_map;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 06:11:34 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/cc/framework/gradients.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 22K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
func_operand_types.reserve(operands.size()); for (Value operand : operands) func_operand_types.push_back(operand.getType()); // Function results are the yield operands SmallVector<Type, 16> func_result_types; for (Value operand : island_op.GetYield().getOperands()) func_result_types.push_back(operand.getType()); FunctionType func_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0)