Searched defs:feature_shape (Results 1 – 4 of 4) sorted by relevance
193 shape_inference::ShapeHandle feature_shape; in __anon8a3fd5e50702() local220 shape_inference::ShapeHandle feature_shape; in __anon8a3fd5e50802() local260 shape_inference::ShapeHandle feature_shape; in __anon8a3fd5e50a02() local402 ShapeHandle feature_shape; in __anon8a3fd5e50f02() local478 ShapeHandle feature_shape; in __anon8a3fd5e51402() local
212 const Shape feature_shape = scale->shape(); in HandleBatchNormTraining() local341 const Shape feature_shape = scale->shape(); in HandleBatchNormInference() local465 const Shape feature_shape = scale->shape(); in HandleBatchNormGrad() local
1527 Shape feature_shape = in InferBatchNormGradShape() local
332 def create_dummy_tensor(feature_shape, feature_type): argument