Searched refs:example_weights_t (Results 1 – 2 of 2) sorted by relevance
153 const Tensor* example_weights_t; in Compute() local155 context->input(kExampleWeightsName, &example_weights_t)); in Compute()157 auto example_weights = example_weights_t->flat<float>(); in Compute()
339 const Tensor* example_weights_t; in Initialize() local340 TF_RETURN_IF_ERROR(context->input("example_weights", &example_weights_t)); in Initialize()341 auto example_weights = example_weights_t->flat<float>(); in Initialize()