Home
last modified time | relevance | path

Searched defs:feature_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dboosted_trees_ops.cc366 shape_inference::ShapeHandle feature_shape; in __anon770f674c0a02() local
418 shape_inference::ShapeHandle feature_shape; in __anon770f674c0b02() local
462 shape_inference::ShapeHandle feature_shape; in __anon770f674c0c02() local
491 shape_inference::ShapeHandle feature_shape; in __anon770f674c0d02() local
533 shape_inference::ShapeHandle feature_shape; in __anon770f674c0f02() local
762 ShapeHandle feature_shape; in __anon770f674c1502() local
852 ShapeHandle feature_shape; in __anon770f674c1b02() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc179 const Shape feature_shape = scale->shape(); in HandleBatchNormTraining() local
308 const Shape feature_shape = scale->shape(); in HandleBatchNormInference() local
414 const Shape feature_shape = scale->shape(); in HandleBatchNormGrad() local
Dshape_inference.cc1571 Shape feature_shape = in InferBatchNormGradShape() local
/external/tensorflow/tensorflow/python/keras/engine/
Ddata_adapter_test.py100 def __init__(self, batch_size, feature_shape): argument
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dstats_ops.cc1558 const auto feature_shape = feature_shape_t->vec<int32>(); in Compute() local