- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 442 for pushBack (0.24 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
// TODO: b/310291615 - find a better way for platform support. arg_types.push_back(RankedTensorType::get({}, builder.getI32Type())); arg_locs.push_back(module_op.getLoc()); for (const Value input_value : inputs) { arg_types.push_back(input_value.getType()); arg_locs.push_back(input_value.getLoc()); } // Identify result types. SmallVector<Type> result_types;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0) -
tensorflow/compiler/jit/device_util.cc
return it->second; } int new_id = names_.size(); names_.push_back(string(name)); id_to_device_type_.push_back(std::make_unique<DeviceType>("")); DeviceType* device_type = id_to_device_type_.back().get(); TF_RETURN_IF_ERROR(DeviceNameToDeviceType(names_.back(), device_type)); is_cpu_.push_back(device_type->type_string() == DEVICE_CPU); is_gpu_.push_back(device_type->type_string() == DEVICE_GPU);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/tests/value_test.cc
TaggedValue valuef(3.f); TaggedValue valuei(int64_t(3)); TaggedValue list = TaggedValue::List(); TaggedValue tuple = TaggedValue::Tuple(); tuple.tuple().push_back(TaggedValue(int64_t(310))); list.list().push_back(valuei); list.list().push_back(valuef); list.list().push_back(tuple); std::stringstream stream; stream << list; ASSERT_EQ(stream.str(), "[3, 3, (310, ), ]"); } TEST(ValueTest, TestString) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 01 11:18:25 UTC 2022 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/representative_dataset_test.cc
*(config_1.mutable_tf_record()->mutable_path()) = "test_path_1"; representative_dataset_configs.push_back(config_1); // Signature key is implicitly "serving_default". RepresentativeDatasetConfig config_2{}; *(config_2.mutable_tf_record()->mutable_path()) = "test_path_2"; representative_dataset_configs.push_back(config_2); const absl::StatusOr< absl::flat_hash_map<std::string, RepresentativeDatasetFile>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 29 04:53:21 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
auto type = std::get<1>(it); if (value.getType() == type) { args.push_back(value); } else { if (IsCompatibleTypeWithTFLCastOp(value.getType()) && IsCompatibleTypeWithTFLCastOp(type)) { auto cast = b.create<CastOp>(yield_op->getLoc(), type, value); args.push_back(cast); } else { auto cast = b.create<TF::CastOp>(yield_op->getLoc(), type, value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/cc/gradients/manip_grad.cc
auto shift = op.input(1); auto axis = op.input(2); auto grad_op = Roll(scope, grad_inputs[0], Neg(scope, shift), axis); grad_outputs->push_back(grad_op); grad_outputs->push_back(NoGradient()); grad_outputs->push_back(NoGradient()); return scope.status(); } REGISTER_GRADIENT_OP("Roll", RollGrad); } // namespace } // namespace ops
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 19 12:19:42 UTC 2020 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
}; TEST_F(PjRtExecutionUtilTest, 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 Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0) -
tensorflow/cc/gradients/functional_grad.cc
for (int i = 0; i < num_inputs; i++) { func_inputs.push_back(op.input(i)); input_dtypes.push_back(op.input_type(i)); } func_inputs.insert(std::end(func_inputs), std::begin(grad_inputs), std::end(grad_inputs)); auto grad = SymbolicGradient(scope, func_inputs, input_dtypes, f); for (int i = 0; i < num_inputs; i++) { grad_outputs->push_back(grad[i]); } return scope.status(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 15 20:09:06 UTC 2021 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc
return WalkResult::interrupt(); } if (llvm::find(func_metadata.inputs, value) == func_metadata.inputs.end()) { func_metadata.inputs.push_back(value); func_metadata.input_devices.push_back(value_device); } continue; } Operation *defining_op = value.getDefiningOp(); std::string defining_op_host = GetHost(defining_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/model/op_spec.cc
output_args_.push_back(arg); } // Parse the attributes. for (const OpDef::AttrDef& attr_def : op_def.attr()) { AttrSpec attr = AttrSpec::Create(attr_def); // Only non-inferred args are added as arguments. if (inferred_attrs.find(attr_def.name()) == inferred_attrs.end()) { argument_attrs_.push_back(attr); } } } } // namespace generator
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 01 21:05:56 UTC 2021 - 2.1K bytes - Viewed (0)