Home
last modified time | relevance | path

Searched defs:shape_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dreshape_test_common.h48 std::initializer_list<int> shape_shape, in ReshapeOpModel()
95 std::initializer_list<int> shape_shape, in BuildWithTensorShape()
115 std::initializer_list<int> shape_shape, in BuildWithConstantTensorShape()
Dbroadcast_to_test.cc33 std::initializer_list<int> shape_shape) { in BroadcastToOpModel()
45 std::initializer_list<int> shape_shape, in BroadcastToOpModel()
Dscatter_nd.cc52 const RuntimeShape& shape_shape, in CheckShapes()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dempty_op.cc43 const TensorShape shape_shape = ctx->InputShape("shape"); in Compile() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_concat_op_test.py37 shape_shape=None): argument
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test.cc923 Output shape_shape = ops::Const(root.WithOpName("shape_shape"), {2}, {1}); in TEST() local
994 Output shape_shape = in TEST() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc2228 ShapeHandle values_shape, ShapeHandle shape_shape) { in ValidateSparseTensor()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc2720 TensorShape shape_shape({var_shape.dims()}); in CreatePerHostDummyArgs() local