Searched refs:take_all (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | filter_fusion_test.py | 37 take_all = lambda x: constant_op.constant(True) function
40 predicates = [take_all, is_zero, greater]
47 take_all = lambda x, y: constant_op.constant(True) function
50 cases.append((lambda x: (x, x), "Tuple1", [take_all, take_all]))
51 cases.append((lambda x: (x, 2), "Tuple2", [take_all, is_zero]))
|
| D | map_and_filter_fusion_test.py | 47 take_all = lambda x: constant_op.constant(True) function
51 predicates = [take_all, is_zero, is_odd, greater]
60 take_all = lambda x, y: constant_op.constant(True) function
62 predicates = [take_all, is_zero]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | slice_op.cc | 113 const bool take_all = (b == 0) && (s == input.dim_size(i)); in SharedSliceValidation() local
114 (*is_identity) &= take_all; in SharedSliceValidation()
115 (*slice_dim0) &= (i == 0) || take_all; in SharedSliceValidation()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 127 const bool take_all = (b == 0) && (s == input_tf_shape.dim_size(i)); in ValidateMklInputs() local
128 (*is_identity) &= take_all; in ValidateMklInputs()
|