Searched defs:logits_shape (Results 1 – 9 of 9) sorted by relevance
124 ShapeHandle logits_shape; in __anon9ce6b60a0302() local
105 ShapeHandle logits_shape; in __anon4f32d8010202() local
477 auto logits_shape = in __anon9348109b0c02() local555 auto logits_shape = in __anon9348109b0f02() local
44 TensorShape logits_shape = ctx->InputShape(0); in Compile() local
129 const TensorShape logits_shape = ctx->InputShape(0); in Compile() local
71 auto logits_shape = {num_batches, num_classes}; in MultinomialOpModel() local
128 TensorShape logits_shape({5, 3}); in TEST_F() local
2537 TensorShapeProto logits_shape = op_context.op_info.inputs(0).shape(); in PredictSoftmax() local
1381 ArrayRef<int64_t> logits_shape = output_type.getShape(); in convertSoftmaxOp() local