MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / Matrix

Method Matrix

tensorflow/core/framework/shape_inference.cc:772–775  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

770}
771
772ShapeHandle InferenceContext::Matrix(DimensionOrConstant dim1,
773 DimensionOrConstant dim2) {
774 return MakeShape({dim1, dim2});
775}
776
777Status InferenceContext::MakeShapeFromShapeTensorTreatScalarAsUnknownShape(
778 int input_idx, ShapeHandle* out) {

Callers 15

model_ops.ccFile · 0.80
routing_function_op.ccFile · 0.80
hard_routing_function_op.ccFile · 0.80
routing_gradient_op.ccFile · 0.80
k_feature_routing_function_op.ccFile · 0.80
stochastic_hard_routing_gradient_op.ccFile · 0.80
unpack_path_op.ccFile · 0.80
stochastic_hard_routing_function_op.ccFile · 0.80
beam_search_ops.ccFile · 0.80
ApplyGradientTreesPredictionShapeFnFunction · 0.80
ApplyGradientTreesPredictionVerboseShapeFnFunction · 0.80
sparse_feature_cross_op.ccFile · 0.80

Calls 1

MakeShapeFunction · 0.85

Tested by 2

TESTFunction · 0.64
TEST_FFunction · 0.64