/external/tensorflow/tensorflow/examples/adding_an_op/ |
H A D | attr_examples.cc | 19 REGISTER_OP("RestrictedTypeExample").Attr("t: {int32, float, bool}"); 21 REGISTER_OP("NumberType").Attr("t: numbertype"); 23 REGISTER_OP("EnumExample").Attr("e: {'apple', 'orange'}"); 25 REGISTER_OP("MinIntExample").Attr("a: int >= 2"); 27 REGISTER_OP("TypeListExample").Attr("a: list({int32, float}) >= 3"); 29 REGISTER_OP("AttrDefaultExample").Attr("i: int = 0"); 32 .Attr("s: string = 'foo'") 33 .Attr("i: int = 0") 34 .Attr("f: float = 1.0") 35 .Attr(" [all...] |
/external/tensorflow/tensorflow/compiler/jit/ops/ |
H A D | xla_ops.cc | 22 .Attr("Tconstants: list(type) >= 0") 24 .Attr("Targs: list(type) >= 0") 26 .Attr("Nresources: int >= 0") 28 .Attr("Tresults: list(type) >= 0") 29 .Attr("function: func")
|
H A D | parallel_check_op.cc | 21 .Attr("T: list(type) >= 0")
|
/external/tensorflow/tensorflow/core/ops/ |
H A D | sendrecv_ops.cc | 23 .Attr("T: type") 24 .Attr("tensor_name: string") 25 .Attr("send_device: string") 26 .Attr("send_device_incarnation: int") 27 .Attr("recv_device: string") 28 .Attr("client_terminated: bool = false") 47 .Attr("tensor_type: type") 48 .Attr("tensor_name: string") 49 .Attr("send_device: string") 50 .Attr("send_device_incarnatio [all...] |
H A D | script_ops.cc | 24 .Attr("token: string") 25 .Attr("Tin: list(type) >= 0") 26 .Attr("Tout: list(type) >=0") 33 .Attr("token: string") 34 .Attr("Tin: list(type) >= 0") 35 .Attr("Tout: list(type) >= 0") 41 .Attr("token: string") 42 .Attr("Tin: list(type) >= 0") 43 .Attr("Tout: list(type) >=0")
|
H A D | word2vec_ops.cc | 32 .Attr("filename: string") 33 .Attr("batch_size: int") 34 .Attr("window_size: int = 5") 35 .Attr("min_count: int = 5") 36 .Attr("subsample: float = 1e-3"); 48 .Attr("vocab_count: list(int)") 49 .Attr("num_negative_samples: int");
|
H A D | manip_ops.cc | 28 .Attr("T: type") 29 .Attr("Tshift: {int32,int64}") 30 .Attr("Taxis: {int32,int64}")
|
H A D | candidate_sampling_ops.cc | 51 .Attr("num_true: int >= 1") 52 .Attr("num_sampled: int >= 1") 53 .Attr("unique: bool") 54 .Attr("range_max: int >= 1") 55 .Attr("seed: int = 0") 56 .Attr("seed2: int = 0") 65 .Attr("num_true: int >= 1") 66 .Attr("num_sampled: int >= 1") 67 .Attr("unique: bool") 68 .Attr("range_ma [all...] |
H A D | function_ops.cc | 23 .Attr("T: type") 24 .Attr("index: int >= 0") 39 .Attr("T: type") 40 .Attr("index: int >= 0") 55 .Attr("Tin: list(type)") 56 .Attr("T: type") 57 .Attr("N: int >= 1") 65 .Attr("T: type") 66 .Attr("N: int >= 1") 67 .Attr("out_type [all...] |
H A D | random_ops.cc | 30 .Attr("seed: int = 0") 31 .Attr("seed2: int = 0") 32 .Attr("dtype: {half,bfloat16,float,double}") 33 .Attr("T: {int32, int64}") 42 .Attr("seed: int = 0") 43 .Attr("seed2: int = 0") 44 .Attr("Tout: {int32, int64}") 45 .Attr("T: {int32, int64}") 52 .Attr("seed: int = 0") 53 .Attr("seed [all...] |
H A D | functional_ops.cc | 27 .Attr("Tin: list(type)") 28 .Attr("Tout: list(type)") 29 .Attr("f: func") 47 .Attr("Tin: list(type)") 48 .Attr("Tout: list(type)") 49 .Attr("f: func") 56 .Attr("Tcond: type") 57 .Attr("Tin: list(type)") 58 .Attr("Tout: list(type)") 59 .Attr("then_branc [all...] |
H A D | logging_ops.cc | 26 .Attr("T: list(type)") 27 .Attr("summarize: int = 3") 35 .Attr("T: type") 36 .Attr("U: list(type) >= 0") 37 .Attr("message: string = ''") 38 .Attr("first_n: int = -1") 39 .Attr("summarize: int = 3") 53 .Attr("T: type") 59 .Attr("T: type") 60 .Attr("descriptio [all...] |
H A D | dataset_ops.cc | 38 .Attr("Toutput_types: list(type) >= 1") 39 .Attr("output_shapes: list(shape) >= 1") 50 .Attr("Toutput_types: list(type) >= 1") 51 .Attr("output_shapes: list(shape) >= 1") 64 .Attr("Tvalues: type") 72 .Attr("output_types: list(type) >= 1") 73 .Attr("output_shapes: list(shape) >= 1") 74 .Attr("N: int >= 1") 81 .Attr("output_types: list(type) >= 1") 82 .Attr("output_shape [all...] |
H A D | state_ops.cc | 26 .Attr("shape: shape") 27 .Attr("dtype: type") 28 .Attr("container: string = ''") 29 .Attr("shared_name: string = ''") 35 .Attr("shape: shape") 36 .Attr("dtype: type") 37 .Attr("container: string = ''") 38 .Attr("shared_name: string = ''") 60 .Attr("dtype: type") 66 .Attr("shap [all...] |
H A D | batch_ops.cc | 27 .Attr("num_batch_threads: int") 28 .Attr("max_batch_size: int") 29 .Attr("max_enqueued_batches: int = 10") 30 .Attr("batch_timeout_micros: int") 31 .Attr("allowed_batch_sizes: list(int) = []") 32 .Attr("grad_timeout_micros: int") 33 .Attr("container: string = ''") 34 .Attr("shared_name: string = ''") 35 .Attr("batching_queue: string = ''") 36 .Attr(" [all...] |
/external/tensorflow/tensorflow/contrib/cloud/ops/ |
H A D | bigquery_reader_ops.cc | 25 .Attr("container: string = ''") 26 .Attr("shared_name: string = ''") 27 .Attr("project_id: string") 28 .Attr("dataset_id: string") 29 .Attr("table_id: string") 30 .Attr("columns: list(string)") 31 .Attr("timestamp_millis: int") 32 .Attr("test_end_point: string = ''") 58 .Attr("project_id: string") 59 .Attr("dataset_i [all...] |
/external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
H A D | functional_ops.cc | 26 .Attr("T: list(type) >= 0") 27 .Attr("cond: func") 28 .Attr("body: func") 53 .Attr("Tcond: type") 54 .Attr("then_branch: func") 55 .Attr("else_branch: func") 56 .Attr("Tin: list(type) >= 0") 57 .Attr("Tout: list(type) >= 0")
|
/external/tensorflow/tensorflow/contrib/tensorrt/ops/ |
H A D | trt_engine_op.cc | 31 .Attr("serialized_engine: string") 32 .Attr("input_nodes: list(string)") 33 .Attr("output_nodes: list(string)") 34 .Attr("InT: list({float32})") 35 .Attr("OutT: list({float32})")
|
/external/tensorflow/tensorflow/user_ops/ |
H A D | invalid_op.cc | 22 .Attr("invalid attr: int32") // invalid since the name has a space.
|
/external/tensorflow/tensorflow/cc/framework/ |
H A D | test_op.cc | 25 .Attr("scope: int") 26 .Attr("builder: int = 1") 27 .Attr("while: int") 38 .Attr("scope: int = 2") 39 .Attr("throw_away2: int = 2") 40 .Attr("attrs: int = 4") 41 .Attr("node: int = 4") 54 .Attr("node: int = 4")
|
/external/tensorflow/tensorflow/contrib/data/ops/ |
H A D | dataset_ops.cc | 23 .Attr("output_types: list(type) >= 1") 24 .Attr("output_shapes: list(shape) >= 1") 34 .Attr("shared_name: string") 35 .Attr("container: string") 36 .Attr("f: func") 37 .Attr("buffer_size: int") 38 .Attr("thread_pool_size: int") 57 .Attr("output_types: list(type)")
|
/external/tensorflow/tensorflow/contrib/nccl/ops/ |
H A D | nccl_ops.cc | 27 .Attr("reduction: {'min', 'max', 'prod', 'sum'}") 28 .Attr("T: {float, float64, int32, int64}") 29 .Attr("num_devices: int") 30 .Attr("shared_name: string") 53 .Attr("reduction: {'min', 'max', 'prod', 'sum'}") 54 .Attr("T: {float, float64, int32, int64}") 55 .Attr("num_devices: int") 71 .Attr("reduction: {'min', 'max', 'prod', 'sum'}") 72 .Attr("T: {float, float64, int32, int64}") 73 .Attr("num_device [all...] |
/external/tensorflow/tensorflow/contrib/image/ops/ |
H A D | single_image_random_dot_stereograms_ops.cc | 27 .Attr("T: {double,float,int64,int32}") 30 .Attr("hidden_surface_removal: bool = true") 31 .Attr("convergence_dots_size: int = 8") 32 .Attr("dots_per_inch: int = 72") 33 .Attr("eye_separation: float = 2.5") 34 .Attr("mu: float = .3333") 35 .Attr("normalize: bool = true") 36 .Attr("normalize_max: float = -100.0") 37 .Attr("normalize_min: float = 100.0") 38 .Attr("border_leve [all...] |
/external/tensorflow/tensorflow/contrib/periodic_resample/ops/ |
H A D | array_ops.cc | 25 .Attr("T: numbertype") 27 .Attr("shape: shape")
|
/external/tensorflow/tensorflow/contrib/tpu/ops/ |
H A D | replication_ops.cc | 26 .Attr("num_replicas: int >= 0") 27 .Attr("topology: string = \"\"") 28 .Attr("device_assignment: list(int) = []") 29 .Attr("computation_shape: list(int) = []") 35 .Attr("N: int >= 1") 36 .Attr("T: type") 54 .Attr("num_replicas: int >= 1") 55 .Attr("T: type") 67 .Attr("computation: func") 68 .Attr("num_replica [all...] |