Home
last modified time | relevance | path

Searched refs:dispatch_for_api (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/python/util/
Ddispatch_test.py351 @dispatch.dispatch_for_api(math_ops.add, {
372 @dispatch.dispatch_for_api(math_ops.add, {"x": MaskedTensor})
399 @dispatch.dispatch_for_api(math_ops.add, {"x": MaskedTensor},
422 @dispatch.dispatch_for_api(array_ops.concat,
444 @dispatch.dispatch_for_api(math_ops.add, {
470 @dispatch.dispatch_for_api(math_ops.add)
499 @dispatch.dispatch_for_api(
520 @dispatch.dispatch_for_api(math_ops.add)
538 @dispatch.dispatch_for_api(math_ops.add, {0: MaskedTensor, 1: MaskedTensor})
556 @dispatch.dispatch_for_api(math_ops.add, {
[all …]
Ddispatch.py251 def dispatch_for_api(api, *signatures): function
980 @dispatch_for_api(api, {x_name: x_type})
1016 @dispatch_for_api(api, {x_name: x_type, y_name: y_type})
1080 _update_docstring_with_api_list(dispatch_for_api,
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_math_ops.py266 @dispatch.dispatch_for_api(math_ops.unsorted_segment_sum)
280 @dispatch.dispatch_for_api(math_ops.unsorted_segment_prod)
294 @dispatch.dispatch_for_api(math_ops.unsorted_segment_min)
308 @dispatch.dispatch_for_api(math_ops.unsorted_segment_max)
322 @dispatch.dispatch_for_api(math_ops.unsorted_segment_mean)
342 @dispatch.dispatch_for_api(math_ops.unsorted_segment_sqrt_n)
609 @dispatch.dispatch_for_api(math_ops.reduce_sum)
625 @dispatch.dispatch_for_api(math_ops.reduce_prod)
640 @dispatch.dispatch_for_api(math_ops.reduce_min)
655 @dispatch.dispatch_for_api(math_ops.reduce_max)
[all …]
Dragged_array_ops.py215 @dispatch.dispatch_for_api(array_ops.tile)
389 @dispatch.dispatch_for_api(array_ops.expand_dims_v2)
462 @dispatch.dispatch_for_api(array_ops.expand_dims)
478 @dispatch.dispatch_for_api(array_ops.size_v2)
503 @dispatch.dispatch_for_api(array_ops.size)
514 @dispatch.dispatch_for_api(array_ops.rank)
544 @dispatch.dispatch_for_api(array_ops.one_hot)
679 @dispatch.dispatch_for_api(array_ops.reverse)
875 @dispatch.dispatch_for_api(data_flow_ops.dynamic_partition)
890 @dispatch.dispatch_for_api(array_ops.split)
[all …]
Dragged_image_ops.py31 @dispatch.dispatch_for_api(image_ops.resize_images_v2)
49 @dispatch.dispatch_for_api(image_ops.resize_images)
Dragged_squeeze_op.py29 @dispatch.dispatch_for_api(array_ops.squeeze_v2)
124 @dispatch.dispatch_for_api(array_ops.squeeze)
Dragged_gather_ops.py33 @dispatch.dispatch_for_api(array_ops.gather_v2)
325 @dispatch.dispatch_for_api(array_ops.gather)
338 @dispatch.dispatch_for_api(array_ops.gather_nd_v2)
481 @dispatch.dispatch_for_api(array_ops.gather_nd)
Dragged_check_ops.py22 @dispatch.dispatch_for_api(check_ops.assert_type)
Dragged_batch_gather_ops.py26 @dispatch.dispatch_for_api(array_ops.batch_gather)
Dragged_where_op.py30 @dispatch.dispatch_for_api(array_ops.where_v2)
95 @dispatch.dispatch_for_api(array_ops.where)
Dragged_concat_ops.py31 @dispatch.dispatch_for_api(array_ops.concat)
74 @dispatch.dispatch_for_api(array_ops.stack)
Dragged_string_ops.py663 @dispatch.dispatch_for_api(string_ops.reduce_join_v2)
830 @dispatch.dispatch_for_api(string_ops.string_format)
942 @dispatch.dispatch_for_api(string_ops.string_join)
/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/
Dmultiplex_3_op.py40 @tf.experimental.dispatch_for_api(examples_multiplex_dense)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.experimental.pbtxt36 name: "dispatch_for_api"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.experimental.pbtxt52 name: "dispatch_for_api"
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_array_ops.py33 @dispatch.dispatch_for_api(array_ops.shape_v2)
41 @dispatch.dispatch_for_api(array_ops.shape)