- Sort Score
- Result 10 results
- Languages All
Results 81 - 86 of 86 for GetShape (0.5 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
if (!ranked_type) return context_op->emitOpError() << "A map_outside_compilation op's input and output types must be " "ranked tensors."; ArrayRef<int64_t> in_shape = ranked_type.getShape(); if (in_shape.empty() || in_shape[0] < 0) { return context_op->emitOpError() << "A map_outside_compilation op's input and output shapes must "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc
auto output_uniform_quantized_tensor_type = RankedTensorType::getChecked( rewriter.getUnknownLoc(), /*shape=*/ mlir::cast<TensorType>(conv_output_value.getType()).getShape(), output_uniform_quantized_type); SmallVector<Type> new_conv_output_types = { output_uniform_quantized_tensor_type}; auto new_conv_op_with_output_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/c/tf_tensor.cc
return static_cast<TF_DataType>(t->tensor->Type()); } void TF_SetShape(TF_Tensor* t, const int64_t* dims, int num_dims) { tensorflow::down_cast<tensorflow::TensorInterface*>(t->tensor)->SetShape( dims, num_dims); } int TF_NumDims(const TF_Tensor* t) { return t->tensor->NumDims(); } int64_t TF_Dim(const TF_Tensor* t, int dim_index) { return t->tensor->Dim(dim_index); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 21:57:32 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference.cc
} shape_inference::ShapeHandle handle; TF_RETURN_IF_ERROR( context->MakeShapeFromPartialTensorShape(arg_shape.shape, &handle)); TF_RETURN_IF_ERROR(shape_refiner->SetShape(n, 0, handle)); } } // Sometimes we have VariableShape nodes in while loop (after Enter nodes). // They won't be constant-folded because TensorFlow constant folding does
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/c/c_api.cc
return; } tensorflow::shape_inference::ShapeHandle new_shape = tensorflow::ShapeHandleFromDims(ic, num_dims, dims); status->status = graph->refiner.SetShape(node, output.index, new_shape); } int TF_GraphGetTensorNumDims(TF_Graph* graph, TF_Output output, TF_Status* status) { Node* node = &output.oper->node;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
TF_Buffer* buf = TF_GetAllOpList(); tensorflow::OpList op_list; EXPECT_TRUE(op_list.ParseFromArray(buf->data, buf->length)); EXPECT_GT(op_list.op_size(), 0); TF_DeleteBuffer(buf); } TEST(CAPI, SetShape) { TF_Status* s = TF_NewStatus(); TF_Graph* graph = TF_NewGraph(); TF_Operation* feed = Placeholder(graph, s); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); TF_Output feed_out_0 = TF_Output{feed, 0};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 96.9K bytes - Viewed (0)