Searched refs:dim_args_spec (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_call_module_op.cc | 47 std::vector<string> dim_args_spec,
206 std::vector<string> dim_args_spec, in PopulateDimArgInputs() argument
209 int nr_dim_args = dim_args_spec.size(); in PopulateDimArgInputs()
211 string dim_arg_spec = dim_args_spec[i]; in PopulateDimArgInputs()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 566 def call_module(args, *, module, Tout, Sout, dim_args_spec=()): argument
568 args, module=module, dim_args_spec=dim_args_spec, Tout=Tout, Sout=Sout)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_call_module_test.py | 106 dim_args_spec=['0.1'])
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 55867 …args []tf.Output, module string, Sout []tf.Shape, Tout []tf.DataType, dim_args_spec []string) (out…
55871 …ap[string]interface{}{"module": module, "Sout": Sout, "Tout": Tout, "dim_args_spec": dim_args_spec}
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 19833 StrArrayAttr:$dim_args_spec
|