Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for add_axis (0.13 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
Back to top