- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 194 for pushBack (0.21 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h
if (mlir::isa<NoneType>(operand_type)) { inputs.push_back(operand); continue; } auto ele_type = mlir::cast<TensorType>(operand.getType()).getElementType(); if (auto dq_op = dyn_cast_or_null<DequantizeOpT>(operand.getDefiningOp())) { inputs.push_back(dq_op.getOperand()); } else if (!ele_type.isF32()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
}; TEST_F(PjRtExecutionUtilGpuTest, PreparePjRtExecutableArguments) { std::vector<const Tensor*> inputs; inputs.push_back(CreateDeviceTensor<int32_t>(TensorShape({1, 3}), {0, 0, 0})); inputs.push_back(CreateDeviceTensor<int32_t>(TensorShape({1, 3}), {1, 2, 3})); inputs.push_back(CreateDeviceTensor<int32_t>(TensorShape({1, 3}), {4, 5, 6})); int num_missing_prefix_ctx_inputs = 2; std::vector<int> input_mapping{3, 4};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K bytes - Viewed (0) -
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/compiler/jit/encapsulate_util.cc
std::vector<const Edge*> control_edges; std::vector<OutEdgeInfo> data_edges; for (auto e : n->out_edges()) { if (e->IsControlEdge()) { control_edges.push_back(e); } else { data_edges.push_back({e->dst(), e->src_output(), e->dst_input()}); } } for (const Edge* e : control_edges) { g->AddControlEdge(original_node, e->dst()); g->RemoveEdge(e); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K 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/tensorflow/transforms/breakup-islands.cc
// Add new control input if its defining op is not already a defining // op for some other operand. Update defining_ops. if (defining_ops.insert(new_control_input.getDefiningOp()).second) { operands.push_back(new_control_input); } } state.addOperands(operands); Operation* new_op = builder.create(state); item.replaceAllUsesWith(new_op); new_op->setAttrs(item.getAttrDictionary());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K bytes - Viewed (0)