- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for add_axis (0.13 sec)
-
tensorflow/compiler/mlir/lite/schema/schema_generated.h
}; struct ConcatenationOptionsBuilder { typedef ConcatenationOptions Table; ::flatbuffers::FlatBufferBuilder &fbb_; ::flatbuffers::uoffset_t start_; void add_axis(int32_t axis) { fbb_.AddElement<int32_t>(ConcatenationOptions::VT_AXIS, axis, 0); } void add_fused_activation_function(tflite::ActivationFunctionType fused_activation_function) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0) -
buildscripts/minio-upgrade.sh
local got got=$(mc cat "$2" | sha256sum) if [ "${expected}" != "${got}" ]; then echo "mismatch - expected ${expected}, got ${got}" exit 1 fi echo "matches - ${expected}, got ${got}" } add_alias() { for i in $(seq 1 4); do echo "... attempting to add alias $i" until (mc alias set minio http://127.0.0.1:9000 minioadmin minioadmin); do echo "...waiting... for 5secs" && sleep 5 done done
Registered: Sun Jun 16 00:44:34 UTC 2024 - Last Modified: Thu Jun 13 05:08:11 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/utils/test_metadata_config.cc
xla_shape.dimensions(), &tensor_shape)); arg_shapes.emplace_back(tensor_shape); DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(input_type, &dtype)); auto metadata_arg = metadata_proto.add_args(); metadata_arg->set_kind(tpu::TPUCompileMetadataProto::Arg::PARAMETER); metadata_arg->set_dtype(dtype); } return absl::OkStatus(); } absl::Status SetupReturnValues(mlir::ModuleOp module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_to_hlo_test.cc
auto client = xla::ClientLibrary::GetOrCreateCompileOnlyClient(platform).value(); std::vector<TensorShape> arg_shapes = {{1}}; TPUCompileMetadataProto metadata_proto; auto arg = metadata_proto.add_args(); arg->set_dtype(DataType::DT_FLOAT); arg->set_kind(TPUCompileMetadataProto::Arg::PARAMETER); metadata_proto.add_retvals(); bool use_tuple_args = true; std::vector<ShardingAndIndex> arg_core_mapping;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/legalize_tf_quant_test.cc
tensorflow::tpu::TPUCompileMetadataProto metadata_proto; // Set up an arg per arg_shape with the specified type. for (int i = 0; i < arg_shapes.size(); ++i) { auto metadata_arg = metadata_proto.add_args(); metadata_arg->set_kind( tensorflow::tpu::TPUCompileMetadataProto::Arg::PARAMETER); metadata_arg->set_dtype(dtype); } // Set up one dummy retval. metadata_proto.add_retvals();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 18:43:55 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_metadata_utils.cc
Type operand_type = operand_type_and_idx.value(); int index = operand_type_and_idx.index(); tensorflow::tpu::TPUCompileMetadataProto::Arg* arg = metadata->add_args(); tensorflow::DataType dtype; tensorflow::Status status = tensorflow::ConvertToDataType(operand_type, &dtype); if (!status.ok()) return op.emitOpError(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.3K bytes - Viewed (0)