- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for input_float (0.1 sec)
-
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass_test.cc
Scope root = Scope::NewRootScope() .ExitOnError() .WithAssignedDevice(kDeviceName) .WithXlaCluster("cluster_0"); Output input_float = ops::Placeholder(root.WithOpName("input_float"), DT_FLOAT); Output input_i64 = ops::Placeholder(root.WithOpName("input_i64"), DT_INT64); Output begin_begin = ops::Placeholder(root.WithOpName("begin_begin"), DT_INT32);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/custom_aggregator_op.cc
// Calculate min/max statistics. const auto input_flat = input_tensor.flat<float>(); Tensor *min_output = nullptr, *max_output = nullptr; OP_REQUIRES_OK(context, context->allocate_output("min", {}, &min_output)); OP_REQUIRES_OK(context, context->allocate_output("max", {}, &max_output)); min_output->scalar<float>().device( context->template eigen_device<CPUDevice>()) = input_flat.minimum(); max_output->scalar<float>().device(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 6.2K bytes - Viewed (0)