Searched defs:input_squared (Results 1 – 3 of 3) sorted by relevance
44 Status validate_arguments(const ITensorInfo *input, const ITensorInfo *input_squared, const ITensor… in validate_arguments()72 void NENormalizationLayerKernel::configure(const ITensor *input, const ITensor *input_squared, ITen… in configure()183 Iterator input_squared(_input_squared, win); in normalize_float() local277 …ormalizationLayerKernel::validate(const ITensorInfo *input, const ITensorInfo *input_squared, cons… in validate()
51 llvm::Value* input_squared = b->CreateFMul(input_clamped, input_clamped); in EmitFastTanh() local
527 auto input_squared = in XLA_TEST_P() local628 auto input_squared = in XLA_TEST_P() local739 auto input_squared = in XLA_TEST_P() local