Laura Kassovic <******@****.***> 1706138984 -0800

Laura Kassovic <******@****.***> 1706138984 -0800

Laura Kassovic <******@****.***> 1701107622 -0800

juanvallejo <******@****.***> 1517847660 -0500

// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package bresource

type Resource[T any] struct {
	name        string
	initializer Initializer[T]
	cfg         ResConfig
	value       T
}

func Should[T any](r *Resource[T], e error) bool {
	return r.cfg.ShouldRetry(e)
}

type ResConfig struct {
	ShouldRetry func(error) bool
	TearDown    func()

namespace {

// Produce a valid graph with a resource-type input.
FunctionDef PassThroughResource() {
  return FunctionDefHelper::Define(
      /*function_name=*/"PassThroughResource",
      /*arg_def=*/{"in: resource"},
      /*ret_def=*/{"out: resource"},
      /*attr_def=*/{},
      /*node_def=*/
      {{{"out"}, "Identity", {"in"}, {{"T", DataType::DT_RESOURCE}}}});
}

node {
  name: "args_0"
  op: "_Arg"
  attr {
    key: "T"
    value {
      type: DT_RESOURCE
    }
  }
  attr {
    key: "index"
    value {
      i: 0
    }
  }
}
node {
  name: "args_1"
  op: "_Arg"
  attr {
    key: "T"
    value {
      type: DT_RESOURCE
    }
  }
  attr {
    key: "index"
    value {
      i: 1
    }
  }
}
node {

 *       minusConfigurations << configurations.anotherConfiguration
 *
 *       //you can add a wb-resource elements; mandatory keys: 'sourcePath', 'deployPath':
 *       //if sourcePath points to non-existing folder it will *not* be added.
 *       resource sourcePath: 'extra/resource', deployPath: 'deployment/resource'
 *
 *       //you can add a wb-property elements; mandatory keys: 'name', 'value':

# mapped correctly.

# CHECK-LABEL: func @main
# CHECK-SAME:  (%{{.*}}: tensor<*xf32>, %[[ARG_1:.*]]: tensor<*x!tf_type.resource<tensor<f32>>>, %[[ARG_2:.*]]: tensor<*x!tf_type.resource<tensor<f32>>>)
# CHECK-SAME:  control_outputs = "var1_add,var2_add"
# CHECK-SAME:  inputs = "arg0,arg1,arg2"
# CHECK-SAME:  outputs = "ret"

func (rl *ResourceList) Storage() *resource.Quantity {
	return rl.Name(ResourceStorage, resource.BinarySI)
}

// Pods returns the list of pods
func (rl *ResourceList) Pods() *resource.Quantity {
	return rl.Name(ResourcePods, resource.DecimalSI)
}

// StorageEphemeral returns the list of ephemeral storage volumes, if any
func (rl *ResourceList) StorageEphemeral() *resource.Quantity {