Home
last modified time | relevance | path

Searched refs:input_features (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_v2_test.py373 input_features = {'aaa': input_a, 'bbb': input_b}
420 embedding_lookup_a = dense_features(input_features)
421 embedding_lookup_b = sequence_features(input_features)
480 input_features = {'inp': input_sparse_tensor}
503 dense_features(input_features)
525 dense_features(input_features)
564 input_features = {'inp': input_sparse_tensor}
613 embedding_lookup = dense_features(input_features)
631 embedding_lookup = dense_features(input_features)
660 input_features = {'inp': input_sparse_tensor}
[all …]
Dfeature_column_test.py258 input_features = {'aaa': input_a, 'bbb': input_b}
300 fc._LazyBuilder(input_features))
302 fc._LazyBuilder(input_features))
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions.cc226 auto input_features = in TryResolvePaddedShapesForTensorCore() local
230 if (input_features == 3 && (output_features == 32 || output_features == 64)) { in TryResolvePaddedShapesForTensorCore()
Dgpu_conv_rewriter.cc562 int64 input_features = rhs->shape().dimensions(input_feature_dimension); in MatchBackwardInput() local
569 CHECK_EQ(input_features % num_groups, 0) in MatchBackwardInput()
/external/tensorflow/tensorflow/python/kernel_tests/
Dparse_single_example_op_test.py165 input_features = {
187 "features": input_features,
196 "features": input_features,
Dparsing_ops_test.py158 input_features = {
181 "features": input_features,
192 "features": input_features,
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dparse_example_dataset_test.py152 input_features = {
173 input_features,
180 input_features,
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2_test.py5720 input_features = {'aaa': input_a, 'bbb': input_b}
5781 fc.FeatureTransformationCache(input_features), None)
5783 fc.FeatureTransformationCache(input_features), None)
5834 input_features = {'aaa': input_a, 'bbb': input_b}
5877 fc.FeatureTransformationCache(input_features), None)
5879 fc.FeatureTransformationCache(input_features), None)
5929 input_features = {
5965 fc.FeatureTransformationCache(input_features), None)
5967 fc.FeatureTransformationCache(input_features), None)
Dfeature_column_test.py5793 input_features = {'aaa': input_a, 'bbb': input_b}
5855 _LazyBuilder(input_features))
5857 _LazyBuilder(input_features))
5898 input_features = {'aaa': input_a, 'bbb': input_b}
5925 input_features, [embedding_column_a, embedding_column_b],
5954 input_features = {
5988 _LazyBuilder(input_features))
5990 _LazyBuilder(input_features))
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc1738 const int64 input_features = lhs.dimensions(dnums.input_feature_dimension()); in InferConvolveShape() local
1759 if (input_features % feature_group_count != 0 || in InferConvolveShape()
1760 input_features / feature_group_count != kernel_input_features) { in InferConvolveShape()
1767 input_features, feature_group_count, kernel_input_features, in InferConvolveShape()