Home
last modified time | relevance | path

Searched refs:take_all (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dfilter_fusion_test.py35 take_all = lambda x: constant_op.constant(True) function
40 filters = [take_all, is_zero, greater]
Dmap_and_filter_fusion_test.py43 take_all = lambda x: constant_op.constant(True) function
49 filters = [take_all, is_zero, is_odd, greater]
/external/tensorflow/tensorflow/core/kernels/
Dslice_op.cc116 const bool take_all = (b == 0) && (s == input.dim_size(i)); in SharedValidation() local
117 (*is_identity) &= take_all; in SharedValidation()
118 (*slice_dim0) &= (i == 0) || take_all; in SharedValidation()
Dmkl_slice_op.cc129 const bool take_all = (b == 0) && (s == input_tf_shape.dim_size(i)); in ValidateMklInputs() local
130 (*is_identity) &= take_all; in ValidateMklInputs()