Searched defs:feature_shape (Results 1 – 4 of 4) sorted by relevance
193 shape_inference::ShapeHandle feature_shape; in __anone3ce78ae0702() local220 shape_inference::ShapeHandle feature_shape; in __anone3ce78ae0802() local260 shape_inference::ShapeHandle feature_shape; in __anone3ce78ae0a02() local402 ShapeHandle feature_shape; in __anone3ce78ae0f02() local478 ShapeHandle feature_shape; in __anone3ce78ae1402() 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